import cv2 import numpy as np import torch # 初始化模型 import torchvision from diffusers import AutoencoderKL, DDIMScheduler from diffusers.pipelines.stable_diffusion import StableDiffusionPipeline from einops import rearrange, repeat from omegaconf import OmegaConf from PIL import Image from torchvision ...