diff options
| author | evshiron <evshiron@gmail.com> | 2022-10-29 19:47:24 +0000 |
|---|---|---|
| committer | evshiron <evshiron@gmail.com> | 2022-10-29 19:49:00 +0000 |
| commit | 7f5212fb5f96092058da4a6f58b1a9037565a00f (patch) | |
| tree | 2fa2ec615e57aaac4a37a6c29f5a36a0ebdf961e /modules/api/models.py | |
| parent | 6b719c49b193a7dfeb64aacfdc8437037e07a2d1 (diff) | |
| parent | d699720254365069866eafcdc519743664075a6d (diff) | |
| download | stable-diffusion-webui-gfx803-7f5212fb5f96092058da4a6f58b1a9037565a00f.tar.gz stable-diffusion-webui-gfx803-7f5212fb5f96092058da4a6f58b1a9037565a00f.tar.bz2 stable-diffusion-webui-gfx803-7f5212fb5f96092058da4a6f58b1a9037565a00f.zip | |
Merge branch 'master' into feat/progress-api
Diffstat (limited to 'modules/api/models.py')
| -rw-r--r-- | modules/api/models.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/modules/api/models.py b/modules/api/models.py index c374a627..8d4abc39 100644 --- a/modules/api/models.py +++ b/modules/api/models.py @@ -1,4 +1,5 @@ import inspect +from click import prompt from pydantic import BaseModel, Field, create_model from typing import Any, Optional from typing_extensions import Literal @@ -150,6 +151,12 @@ class ExtrasBatchImagesRequest(ExtrasBaseRequest): class ExtrasBatchImagesResponse(ExtraBaseResponse): images: list[str] = Field(title="Images", description="The generated images in base64 format.") +class PNGInfoRequest(BaseModel): + image: str = Field(title="Image", description="The base64 encoded PNG image") + +class PNGInfoResponse(BaseModel): + info: str = Field(title="Image info", description="A string with all the info the image had") + class ProgressResponse(BaseModel): progress: float eta_relative: float |