Commit f62db4d5 authored by evshiron's avatar evshiron
Browse files

fix progress response model

parent 7f5212fb
Loading
Loading
Loading
Loading
+0 −30
Original line number Diff line number Diff line
# import time

# from modules.api.models import StableDiffusionTxt2ImgProcessingAPI, StableDiffusionImg2ImgProcessingAPI
# from modules.processing import StableDiffusionProcessingTxt2Img, StableDiffusionProcessingImg2Img, process_images
# from modules.sd_samplers import all_samplers
# from modules.extras import run_pnginfo
# import modules.shared as shared
# from modules import devices
# import uvicorn
# from fastapi import Body, APIRouter, HTTPException
# from fastapi.responses import JSONResponse
# from pydantic import BaseModel, Field, Json
# from typing import List
# import json
# import io
# import base64
# from PIL import Image

# sampler_to_index = lambda name: next(filter(lambda row: name.lower() == row[1].name.lower(), enumerate(all_samplers)), None)

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

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

import time
import uvicorn
from gradio.processing_utils import encode_pil_to_base64, decode_base64_to_file, decode_base64_to_image
+4 −4
Original line number Diff line number Diff line
import inspect
from click import prompt
from pydantic import BaseModel, Field, create_model
from pydantic import BaseModel, Field, Json, create_model
from typing import Any, Optional
from typing_extensions import Literal
from inflection import underscore
@@ -158,6 +158,6 @@ class PNGInfoResponse(BaseModel):
    info: str = Field(title="Image info", description="A string with all the info the image had")

class ProgressResponse(BaseModel):
    progress: float
    eta_relative: float
    state: dict
    progress: float = Field(title="Progress", description="The progress with a range of 0 to 1")
    eta_relative: float = Field(title="ETA in secs")
    state: Json