aboutsummaryrefslogtreecommitdiff
path: root/modules/api/models.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-07-25 08:18:02 +0300
committerAUTOMATIC1111 <16777216c@gmail.com>2023-07-25 08:18:02 +0300
commita3ddf464a2ed24c999f67ddfef7969f8291567be (patch)
treecf70006b4d1d6df1f42ea944416b1034ae32a92b /modules/api/models.py
parentf865d3e11647dfd6c7b2cdf90dde24680e58acd8 (diff)
parent2c11e9009ea18bab4ce2963d44db0c6fd3227370 (diff)
Merge branch 'release_candidate'
Diffstat (limited to 'modules/api/models.py')
-rw-r--r--modules/api/models.py14
1 files changed, 6 insertions, 8 deletions
diff --git a/modules/api/models.py b/modules/api/models.py
index b3a745f0..bf97b1a3 100644
--- a/modules/api/models.py
+++ b/modules/api/models.py
@@ -1,4 +1,5 @@
import inspect
+
from pydantic import BaseModel, Field, create_model
from typing import Any, Optional
from typing_extensions import Literal
@@ -207,11 +208,12 @@ class PreprocessResponse(BaseModel):
fields = {}
for key, metadata in opts.data_labels.items():
value = opts.data.get(key)
- optType = opts.typemap.get(type(metadata.default), type(value))
+ optType = opts.typemap.get(type(metadata.default), type(metadata.default))
- if (metadata is not None):
- fields.update({key: (Optional[optType], Field(
- default=metadata.default ,description=metadata.label))})
+ if metadata.default is None:
+ 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())})
@@ -274,10 +276,6 @@ class PromptStyleItem(BaseModel):
prompt: Optional[str] = Field(title="Prompt")
negative_prompt: Optional[str] = Field(title="Negative Prompt")
-class ArtistItem(BaseModel):
- name: str = Field(title="Name")
- score: float = Field(title="Score")
- category: str = Field(title="Category")
class EmbeddingItem(BaseModel):
step: Optional[int] = Field(title="Step", description="The number of steps that were used to train this embedding, if available")