diff options
author | evshiron <evshiron@gmail.com> | 2022-11-05 14:12:20 +0000 |
---|---|---|
committer | evshiron <evshiron@gmail.com> | 2022-11-05 14:12:20 +0000 |
commit | b6cfaaa20b81d16a978524c952412cfc4d89a414 (patch) | |
tree | e0aa306e34e796c79f657d2a73f558de46f80764 /modules/api/models.py | |
parent | 73e1cd6f53f3566973805666a4049450cdb8da1b (diff) | |
parent | 62e3d71aa778928d63cab81d9d8cde33e55bebb3 (diff) | |
download | stable-diffusion-webui-gfx803-b6cfaaa20b81d16a978524c952412cfc4d89a414.tar.gz stable-diffusion-webui-gfx803-b6cfaaa20b81d16a978524c952412cfc4d89a414.tar.bz2 stable-diffusion-webui-gfx803-b6cfaaa20b81d16a978524c952412cfc4d89a414.zip |
Merge branch 'master' into fix/encode-pnginfo
Diffstat (limited to 'modules/api/models.py')
-rw-r--r-- | modules/api/models.py | 26 |
1 files changed, 13 insertions, 13 deletions
diff --git a/modules/api/models.py b/modules/api/models.py index 2ae75f43..a44c5ddd 100644 --- a/modules/api/models.py +++ b/modules/api/models.py @@ -1,6 +1,6 @@ import inspect from pydantic import BaseModel, Field, create_model -from typing import Any, Optional, Union +from typing import Any, Optional from typing_extensions import Literal from inflection import underscore from modules.processing import StableDiffusionProcessingTxt2Img, StableDiffusionProcessingImg2Img @@ -185,22 +185,22 @@ _options = vars(parser)['_option_string_actions'] for key in _options: if(_options[key].dest != 'help'): flag = _options[key] - _type = str - if(_options[key].default != None): _type = type(_options[key].default) + _type = str + if _options[key].default is not None: _type = type(_options[key].default) flags.update({flag.dest: (_type,Field(default=flag.default, description=flag.help))}) FlagsModel = create_model("Flags", **flags) class SamplerItem(BaseModel): name: str = Field(title="Name") - aliases: list[str] = Field(title="Aliases") + aliases: list[str] = Field(title="Aliases") options: dict[str, str] = Field(title="Options") class UpscalerItem(BaseModel): name: str = Field(title="Name") - model_name: str | None = Field(title="Model Name") - model_path: str | None = Field(title="Path") - model_url: str | None = Field(title="URL") + model_name: Optional[str] = Field(title="Model Name") + model_path: Optional[str] = Field(title="Path") + model_url: Optional[str] = Field(title="URL") class SDModelItem(BaseModel): title: str = Field(title="Title") @@ -211,21 +211,21 @@ class SDModelItem(BaseModel): class HypernetworkItem(BaseModel): name: str = Field(title="Name") - path: str | None = Field(title="Path") + path: Optional[str] = Field(title="Path") class FaceRestorerItem(BaseModel): name: str = Field(title="Name") - cmd_dir: str | None = Field(title="Path") + cmd_dir: Optional[str] = Field(title="Path") class RealesrganItem(BaseModel): name: str = Field(title="Name") - path: str | None = Field(title="Path") - scale: int | None = Field(title="Scale") + path: Optional[str] = Field(title="Path") + scale: Optional[int] = Field(title="Scale") class PromptStyleItem(BaseModel): name: str = Field(title="Name") - prompt: str | None = Field(title="Prompt") - negative_prompt: str | None = Field(title="Negative Prompt") + prompt: Optional[str] = Field(title="Prompt") + negative_prompt: Optional[str] = Field(title="Negative Prompt") class ArtistItem(BaseModel): name: str = Field(title="Name") |