diff options
author | Bruno Seoane <breyxxi@gmail.com> | 2022-11-05 18:56:41 +0000 |
---|---|---|
committer | Bruno Seoane <breyxxi@gmail.com> | 2022-11-05 18:56:41 +0000 |
commit | 59ec427dff84365588a51eb6454f3be47b74fea1 (patch) | |
tree | b0e4a5eef633e4ec9f3f663f45752058d2bba716 /modules/api/models.py | |
parent | fd66f669ea25bad1409aec87ef14b8417009bddc (diff) | |
parent | b08698a09a257365238e43cc9023ce7cf9af73c4 (diff) | |
download | stable-diffusion-webui-gfx803-59ec427dff84365588a51eb6454f3be47b74fea1.tar.gz stable-diffusion-webui-gfx803-59ec427dff84365588a51eb6454f3be47b74fea1.tar.bz2 stable-diffusion-webui-gfx803-59ec427dff84365588a51eb6454f3be47b74fea1.zip |
Merge remote-tracking branch 'upstream/master'
Diffstat (limited to 'modules/api/models.py')
-rw-r--r-- | modules/api/models.py | 32 |
1 files changed, 16 insertions, 16 deletions
diff --git a/modules/api/models.py b/modules/api/models.py index 2ae75f43..f89da1ff 100644 --- a/modules/api/models.py +++ b/modules/api/models.py @@ -1,11 +1,11 @@ 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 from modules.shared import sd_upscalers, opts, parser -from typing import List +from typing import Dict, List API_NOT_ALLOWED = [ "self", @@ -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") - options: dict[str, str] = Field(title="Options") + 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,23 +211,23 @@ 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") score: float = Field(title="Score") - category: str = Field(title="Category")
\ No newline at end of file + category: str = Field(title="Category") |