diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-18 08:48:40 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-18 08:48:40 +0000 |
| commit | 420cc8f68e6aca8a3a0f42ee0e626a6b03712763 (patch) | |
| tree | 6d0003ace94be3bc95009900b800ea494b2688ac | |
| parent | 6be5ccb530dbaac53228e24c69e30909aaa14748 (diff) | |
| download | stable-diffusion-webui-gfx803-420cc8f68e6aca8a3a0f42ee0e626a6b03712763.tar.gz stable-diffusion-webui-gfx803-420cc8f68e6aca8a3a0f42ee0e626a6b03712763.tar.bz2 stable-diffusion-webui-gfx803-420cc8f68e6aca8a3a0f42ee0e626a6b03712763.zip | |
also make None a valid option for options API for #11854
| -rw-r--r-- | modules/api/models.py | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/modules/api/models.py b/modules/api/models.py index b5683071..b55fa728 100644 --- a/modules/api/models.py +++ b/modules/api/models.py @@ -1,4 +1,6 @@ import inspect +import types + from pydantic import BaseModel, Field, create_model from typing import Any, Optional from typing_extensions import Literal @@ -207,11 +209,14 @@ 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)) - if (metadata is not None): - fields.update({key: (Optional[optType], Field( - default=metadata.default ,description=metadata.label))}) + if optType == types.NoneType: + pass + elif metadata is not None: + fields.update({key: (Optional[optType], Field(default=metadata.default, description=metadata.label))}) else: fields.update({key: (Optional[optType], Field())}) |