aboutsummaryrefslogtreecommitdiff
path: root/modules/api/models.py
diff options
context:
space:
mode:
authorxucj98 <975114697@qq.com>2022-11-25 17:07:00 +0800
committerGitHub <noreply@github.com>2022-11-25 17:07:00 +0800
commit263b323de12eb2444b0818105575a2bc69ab0344 (patch)
tree7bc4a7802bdaba21550ad281cc08c3fc1037b074 /modules/api/models.py
parentd20dbe47e06de7f6c0e65242a04c9bb1410ef7cb (diff)
parent828438b4a190759807f9054932cae3a8b880ddf1 (diff)
Merge branch 'AUTOMATIC1111:master' into draft
Diffstat (limited to 'modules/api/models.py')
-rw-r--r--modules/api/models.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/modules/api/models.py b/modules/api/models.py
index f9cd929e..f77951fc 100644
--- a/modules/api/models.py
+++ b/modules/api/models.py
@@ -176,9 +176,9 @@ class InterrogateResponse(BaseModel):
caption: str = Field(default=None, title="Caption", description="The generated caption for the image.")
fields = {}
-for key, value in opts.data.items():
- metadata = opts.data_labels.get(key)
- optType = opts.typemap.get(type(value), type(value))
+for key, metadata in opts.data_labels.items():
+ value = opts.data.get(key)
+ optType = opts.typemap.get(type(metadata.default), type(value))
if (metadata is not None):
fields.update({key: (Optional[optType], Field(