diff options
| author | Vladimir Mandic <mandic00@live.com> | 2023-03-03 14:00:52 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-03-03 14:00:52 +0000 |
| commit | f8e219bad9f33cde94cd31fff3edd70946612541 (patch) | |
| tree | b460967ac68588002166e5f6f3e100993d920bee /modules/api/models.py | |
| parent | 23d4fb5bf2400622d00ca5fe489fadb160ee7c47 (diff) | |
| download | stable-diffusion-webui-gfx803-f8e219bad9f33cde94cd31fff3edd70946612541.tar.gz stable-diffusion-webui-gfx803-f8e219bad9f33cde94cd31fff3edd70946612541.tar.bz2 stable-diffusion-webui-gfx803-f8e219bad9f33cde94cd31fff3edd70946612541.zip | |
allow api requests to specify do not send images in response
Diffstat (limited to 'modules/api/models.py')
| -rw-r--r-- | modules/api/models.py | 18 |
1 files changed, 16 insertions, 2 deletions
diff --git a/modules/api/models.py b/modules/api/models.py index a947e6ac..aa4ea5d5 100644 --- a/modules/api/models.py +++ b/modules/api/models.py @@ -100,13 +100,27 @@ 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_images", "type": bool, "default": False} + ] ).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_images", "type": bool, "default": False} + ] ).generate_model() class TextToImageResponse(BaseModel): |