diff options
| author | Stephen <farrenzo47@gmail.com> | 2022-10-21 23:27:40 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-23 06:26:56 +0000 |
| commit | a7c213d0f5ebb10722629b8490a5863f9ce6c4fa (patch) | |
| tree | 5a6b3accda2399930868f1db71c15a91d4521bc2 /modules/api/processing.py | |
| parent | 1fbfc052eb529d8cf8ce5baf578bcf93d0280c29 (diff) | |
| download | stable-diffusion-webui-gfx803-a7c213d0f5ebb10722629b8490a5863f9ce6c4fa.tar.gz stable-diffusion-webui-gfx803-a7c213d0f5ebb10722629b8490a5863f9ce6c4fa.tar.bz2 stable-diffusion-webui-gfx803-a7c213d0f5ebb10722629b8490a5863f9ce6c4fa.zip | |
[API][Feature] - Add img2img API endpoint
Diffstat (limited to 'modules/api/processing.py')
| -rw-r--r-- | modules/api/processing.py | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/modules/api/processing.py b/modules/api/processing.py index 4c541241..9f1d65c0 100644 --- a/modules/api/processing.py +++ b/modules/api/processing.py @@ -1,7 +1,8 @@ +from array import array from inflection import underscore from typing import Any, Dict, Optional from pydantic import BaseModel, Field, create_model -from modules.processing import StableDiffusionProcessingTxt2Img +from modules.processing import StableDiffusionProcessingTxt2Img, StableDiffusionProcessingImg2Img import inspect @@ -92,8 +93,14 @@ class PydanticModelGenerator: DynamicModel.__config__.allow_mutation = True return DynamicModel -StableDiffusionProcessingAPI = PydanticModelGenerator( +StableDiffusionTxt2ImgProcessingAPI = PydanticModelGenerator( "StableDiffusionProcessingTxt2Img", StableDiffusionProcessingTxt2Img, [{"key": "sampler_index", "type": str, "default": "Euler"}] +).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}] ).generate_model()
\ No newline at end of file |