diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-03-11 10:03:23 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-03-11 10:03:23 +0000 |
| commit | bb3ecc32856dc1341f2fb6c41efca8ade0f5217e (patch) | |
| tree | 6571c30590d5df355e36e02d8e47477617e6aa3b /modules/api/models.py | |
| parent | d81c50391823aaa264bf9c0896a1552e5af5fea6 (diff) | |
| parent | b012d70f15641d6b85c9257b83cec892e941609c (diff) | |
| download | stable-diffusion-webui-gfx803-bb3ecc32856dc1341f2fb6c41efca8ade0f5217e.tar.gz stable-diffusion-webui-gfx803-bb3ecc32856dc1341f2fb6c41efca8ade0f5217e.tar.bz2 stable-diffusion-webui-gfx803-bb3ecc32856dc1341f2fb6c41efca8ade0f5217e.zip | |
Merge pull request #8287 from vladmandic/save-images
Allow saving of images generated via API
Diffstat (limited to 'modules/api/models.py')
| -rw-r--r-- | modules/api/models.py | 24 |
1 files changed, 20 insertions, 4 deletions
diff --git a/modules/api/models.py b/modules/api/models.py index cba43d3b..2b66e1f0 100644 --- a/modules/api/models.py +++ b/modules/api/models.py @@ -14,8 +14,8 @@ API_NOT_ALLOWED = [ "outpath_samples", "outpath_grids", "sampler_index", - "do_not_save_samples", - "do_not_save_grid", + # "do_not_save_samples", + # "do_not_save_grid", "extra_generation_params", "overlay_images", "do_not_reload_embeddings", @@ -100,13 +100,29 @@ class PydanticModelGenerator: StableDiffusionTxt2ImgProcessingAPI = PydanticModelGenerator( "StableDiffusionProcessingTxt2Img", StableDiffusionProcessingTxt2Img, - [{"key": "sampler_index", "type": str, "default": "Euler"}, {"key": "script_name", "type": str, "default": None}, {"key": "script_args", "type": list, "default": []}] + [ + {"key": "sampler_index", "type": str, "default": "Euler"}, + {"key": "script_name", "type": str, "default": None}, + {"key": "script_args", "type": list, "default": []}, + {"key": "do_not_send", "type": bool, "default": False}, + {"key": "do_not_save", "type": bool, "default": True} + ] ).generate_model() StableDiffusionImg2ImgProcessingAPI = PydanticModelGenerator( "StableDiffusionProcessingImg2Img", StableDiffusionProcessingImg2Img, - [{"key": "sampler_index", "type": str, "default": "Euler"}, {"key": "init_images", "type": list, "default": None}, {"key": "denoising_strength", "type": float, "default": 0.75}, {"key": "mask", "type": str, "default": None}, {"key": "include_init_images", "type": bool, "default": False, "exclude" : True}, {"key": "script_name", "type": str, "default": None}, {"key": "script_args", "type": list, "default": []}] + [ + {"key": "sampler_index", "type": str, "default": "Euler"}, + {"key": "init_images", "type": list, "default": None}, + {"key": "denoising_strength", "type": float, "default": 0.75}, + {"key": "mask", "type": str, "default": None}, + {"key": "include_init_images", "type": bool, "default": False, "exclude" : True}, + {"key": "script_name", "type": str, "default": None}, + {"key": "script_args", "type": list, "default": []}, + {"key": "do_not_send", "type": bool, "default": False}, + {"key": "do_not_save", "type": bool, "default": True} + ] ).generate_model() class TextToImageResponse(BaseModel): |