Skip to content

Commit b2c4809

Browse files
committed
fixed api compatibility with python 3.8
1 parent d98eace commit b2c4809

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

modules/api/models.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
from inflection import underscore
77
from modules.processing import StableDiffusionProcessingTxt2Img, StableDiffusionProcessingImg2Img
88
from modules.shared import sd_upscalers
9+
from typing import List
910

1011
API_NOT_ALLOWED = [
1112
"self",
@@ -109,12 +110,12 @@ def generate_model(self):
109110
).generate_model()
110111

111112
class TextToImageResponse(BaseModel):
112-
images: list[str] = Field(default=None, title="Image", description="The generated image in base64 format.")
113+
images: List[str] = Field(default=None, title="Image", description="The generated image in base64 format.")
113114
parameters: dict
114115
info: str
115116

116117
class ImageToImageResponse(BaseModel):
117-
images: list[str] = Field(default=None, title="Image", description="The generated image in base64 format.")
118+
images: List[str] = Field(default=None, title="Image", description="The generated image in base64 format.")
118119
parameters: dict
119120
info: str
120121

@@ -146,10 +147,10 @@ class FileData(BaseModel):
146147
name: str = Field(title="File name")
147148

148149
class ExtrasBatchImagesRequest(ExtrasBaseRequest):
149-
imageList: list[FileData] = Field(title="Images", description="List of images to work on. Must be Base64 strings")
150+
imageList: List[FileData] = Field(title="Images", description="List of images to work on. Must be Base64 strings")
150151

151152
class ExtrasBatchImagesResponse(ExtraBaseResponse):
152-
images: list[str] = Field(title="Images", description="The generated images in base64 format.")
153+
images: List[str] = Field(title="Images", description="The generated images in base64 format.")
153154

154155
class PNGInfoRequest(BaseModel):
155156
image: str = Field(title="Image", description="The base64 encoded PNG image")

0 commit comments

Comments
 (0)