Hydrc68/Yso

1from diffusers import AutoencoderKL, UNet2DConditionModel, StableDiffusionPipeline, StableDiffusionImg2ImgPipeline, DPMSolverMultistepScheduler2import gradio as gr3import torch4from PIL import Image5import utils6import datetime7import time8import psutil9import random10 11 12start_time = time.time()13is_colab = utils.is_google_colab()14state = None15current_steps = 2516 17class Model:18    def __init__(self, name, path="", prefix=""):19        self.name = name20        self.path = path21        self.prefix = prefix22        self.pipe_t2i = None23        self.pipe_i2i = None