Skip to content

Commit ebce0c5

Browse files
committed
Use typing.Optional instead of | to add support for Python 3.9 and below.
1 parent 30b1bcc commit ebce0c5

File tree

1 file changed

+13
-13
lines changed

1 file changed

+13
-13
lines changed

modules/api/models.py

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import inspect
22
from pydantic import BaseModel, Field, create_model
3-
from typing import Any, Optional, Union
3+
from typing import Any, Optional
44
from typing_extensions import Literal
55
from inflection import underscore
66
from modules.processing import StableDiffusionProcessingTxt2Img, StableDiffusionProcessingImg2Img
@@ -185,22 +185,22 @@ class ProgressResponse(BaseModel):
185185
for key in _options:
186186
if(_options[key].dest != 'help'):
187187
flag = _options[key]
188-
_type = str
189-
if(_options[key].default != None): _type = type(_options[key].default)
188+
_type = str
189+
if _options[key].default is not None: _type = type(_options[key].default)
190190
flags.update({flag.dest: (_type,Field(default=flag.default, description=flag.help))})
191191

192192
FlagsModel = create_model("Flags", **flags)
193193

194194
class SamplerItem(BaseModel):
195195
name: str = Field(title="Name")
196-
aliases: list[str] = Field(title="Aliases")
196+
aliases: list[str] = Field(title="Aliases")
197197
options: dict[str, str] = Field(title="Options")
198198

199199
class UpscalerItem(BaseModel):
200200
name: str = Field(title="Name")
201-
model_name: str | None = Field(title="Model Name")
202-
model_path: str | None = Field(title="Path")
203-
model_url: str | None = Field(title="URL")
201+
model_name: Optional[str] = Field(title="Model Name")
202+
model_path: Optional[str] = Field(title="Path")
203+
model_url: Optional[str] = Field(title="URL")
204204

205205
class SDModelItem(BaseModel):
206206
title: str = Field(title="Title")
@@ -211,21 +211,21 @@ class SDModelItem(BaseModel):
211211

212212
class HypernetworkItem(BaseModel):
213213
name: str = Field(title="Name")
214-
path: str | None = Field(title="Path")
214+
path: Optional[str] = Field(title="Path")
215215

216216
class FaceRestorerItem(BaseModel):
217217
name: str = Field(title="Name")
218-
cmd_dir: str | None = Field(title="Path")
218+
cmd_dir: Optional[str] = Field(title="Path")
219219

220220
class RealesrganItem(BaseModel):
221221
name: str = Field(title="Name")
222-
path: str | None = Field(title="Path")
223-
scale: int | None = Field(title="Scale")
222+
path: Optional[str] = Field(title="Path")
223+
scale: Optional[int] = Field(title="Scale")
224224

225225
class PromptStyleItem(BaseModel):
226226
name: str = Field(title="Name")
227-
prompt: str | None = Field(title="Prompt")
228-
negative_prompt: str | None = Field(title="Negative Prompt")
227+
prompt: Optional[str] = Field(title="Prompt")
228+
negative_prompt: Optional[str] = Field(title="Negative Prompt")
229229

230230
class ArtistItem(BaseModel):
231231
name: str = Field(title="Name")

0 commit comments

Comments
 (0)