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