Skip to content

Instantly share code, notes, and snippets.

CONVERTED_MODEL_PATH = "path to converted v2 depth model"
DEVICE = "cuda"
INIT_IMAGE = "path to init image or None"
DEPTH_IMAGE = "path to black and white depth map"
OUTPUT_PATH = "path to save the result"
import diffusers
from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput
import torch