File size: 1,399 Bytes
dd06d6b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
import argparse
from diffusers import DDIMScheduler, DDPMScheduler, UniPCMultistepScheduler
def get_control_signal_type(controlnet):
if "normal" in controlnet:
return "normal"
elif "depth" in controlnet:
return "depth"
else:
raise NotImplementedError
SD_MODELS = {
"sd15_old": "runwayml/stable-diffusion-inpainting",
"sd15_new": "runwayml/stable-diffusion-inpainting",
"sd21": "stabilityai/stable-diffusion-2-inpainting",
"sdxl": "stabilityai/stable-diffusion-xl-base-1.0",
"sdxl_fast": "stabilityai/stable-diffusion-xl-base-1.0",
"sdxl_turbo": "stabilityai/sdxl-turbo",
"sd15_depth": "runwayml/stable-diffusion-inpainting",
}
VAE_MODELS = {
"sdxl": "madebyollin/sdxl-vae-fp16-fix",
"sdxl_fast": "madebyollin/sdxl-vae-fp16-fix",
}
CONTROLNET_MODELS = {
"sd15_old": "fusing/stable-diffusion-v1-5-controlnet-normal",
"sd15_new": "lllyasviel/control_v11p_sd15_normalbae",
"sd21": "thibaud/controlnet-sd21-normalbae-diffusers",
"sdxl": "diffusers/controlnet-depth-sdxl-1.0",
"sdxl_fast": "diffusers/controlnet-depth-sdxl-1.0-small",
"sdxl_turbo": "diffusers/controlnet-depth-sdxl-1.0-small",
"sd15_depth": "lllyasviel/control_v11f1p_sd15_depth",
}
SAMPLERS = {
"ddim": DDIMScheduler,
"ddpm": DDPMScheduler,
"unipc": UniPCMultistepScheduler,
}
DEPTH_ESTIMATOR = "Intel/dpt-hybrid-midas" |