| from diffusers import AutoencoderKL, UNet2DConditionModel, StableDiffusionPipeline, StableDiffusionImg2ImgPipeline, DPMSolverMultistepScheduler | |
| import gradio as gr | |
| import torch | |
| from PIL import Image | |
| import utils | |
| import datetime | |
| import time | |
| import psutil | |
| import random | |
| start_time = time.time() | |
| is_colab = utils.is_google_colab() | |
| state = None | |
| current_steps = 25 | |
| class Model: | |
| def __init__(self, name, path="", prefix=""): | |
| self.name = name | |
| self.path = path | |
| self.prefix = prefix | |
| self.pipe_t2i = None | |
| self.pipe_i2i = None |