diff options
| author | Jabasukuriputo Wang <wfjsw@users.noreply.github.com> | 2023-07-19 02:33:31 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-07-19 02:33:31 +0000 |
| commit | fc3bdf8c1184de1116fe9f30b72905b4b3285e74 (patch) | |
| tree | 64e993c6bce48ddd4fb2d60fd5105bd6c725d52c /modules/api/models.py | |
| parent | 3c570421d3a2eb24528b5f5bb615dcb0c7717e4a (diff) | |
| parent | b010eea520caa90d2a31d98ec7c4a9d9d540c9ad (diff) | |
| download | stable-diffusion-webui-gfx803-fc3bdf8c1184de1116fe9f30b72905b4b3285e74.tar.gz stable-diffusion-webui-gfx803-fc3bdf8c1184de1116fe9f30b72905b4b3285e74.tar.bz2 stable-diffusion-webui-gfx803-fc3bdf8c1184de1116fe9f30b72905b4b3285e74.zip | |
Merge branch 'dev' into move-start-timer
Diffstat (limited to 'modules/api/models.py')
| -rw-r--r-- | modules/api/models.py | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/modules/api/models.py b/modules/api/models.py index 96cfe920..bf97b1a3 100644 --- a/modules/api/models.py +++ b/modules/api/models.py @@ -1,5 +1,4 @@ import inspect -import types from pydantic import BaseModel, Field, create_model from typing import Any, Optional @@ -209,11 +208,9 @@ class PreprocessResponse(BaseModel): fields = {} for key, metadata in opts.data_labels.items(): value = opts.data.get(key) - if key == 'sd_model_checkpoint': - value = None - optType = opts.typemap.get(type(metadata.default), type(value)) + optType = opts.typemap.get(type(metadata.default), type(metadata.default)) - if isinstance(optType, types.NoneType): + if metadata.default is None: pass elif metadata is not None: fields.update({key: (Optional[optType], Field(default=metadata.default, description=metadata.label))}) |