Skip to content

Commit 9b2ca18

Browse files
Merge pull request #4245 from 7flash/7flash/fix-api-compatibility
fixed api compatibility with python 3.8
2 parents 2913b9f + e9c767d commit 9b2ca18

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
@@ -5,6 +5,7 @@
55
from inflection import underscore
66
from modules.processing import StableDiffusionProcessingTxt2Img, StableDiffusionProcessingImg2Img
77
from modules.shared import sd_upscalers, opts, parser
8+
from typing import List
89

910
API_NOT_ALLOWED = [
1011
"self",
@@ -108,12 +109,12 @@ def generate_model(self):
108109
).generate_model()
109110

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

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

@@ -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)