fixed api compatibility with python 3.8 (b2c48091) · Commits · github_fork / Stable Diffusion Webui

modules/api/models.py

+5 −4

Original line number	Diff line number	Diff line
		@@ -6,6 +6,7 @@ from typing_extensions import Literal
		from inflection import underscore
		from modules.processing import StableDiffusionProcessingTxt2Img, StableDiffusionProcessingImg2Img
		from modules.shared import sd_upscalers
		from typing import List

		API_NOT_ALLOWED = [
		"self",
		@@ -109,12 +110,12 @@ StableDiffusionImg2ImgProcessingAPI = PydanticModelGenerator(
		).generate_model()

		class TextToImageResponse(BaseModel):
		images: list[str] = Field(default=None, title="Image", description="The generated image in base64 format.")
		images: List[str] = Field(default=None, title="Image", description="The generated image in base64 format.")
		parameters: dict
		info: str

		class ImageToImageResponse(BaseModel):
		images: list[str] = Field(default=None, title="Image", description="The generated image in base64 format.")
		images: List[str] = Field(default=None, title="Image", description="The generated image in base64 format.")
		parameters: dict
		info: str

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

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

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

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

Admin message