- "inpaint": StableDiffusionXLPAGInpaintPipeline,
- "controlnet": StableDiffusionXLControlNetPAGPipeline,
- # "adapter": StableDiffusionXLAdapterPAGPipeline,
- "controlnet_img2img": StableDiffusionXLControlNetPAGImg2ImgPipeline,
- # "controlnet_inpaint": StableDiffusionXLControlNetPAGInpaintPipeline,
- # "controlnet_union+": StableDiffusionXLControlNetUnionPAGPipeline,
- # "controlnet_union+_inpaint": StableDiffusionXLControlNetUnionPAGInpaintPipeline,
- },
- }
- CONTROLNET_MODEL_IDS = {
- "openpose": ["lllyasviel/control_v11p_sd15_openpose", "r3gm/controlnet-openpose-sdxl-1.0-fp16"],
- "canny": ["lllyasviel/control_v11p_sd15_canny", "r3gm/controlnet-canny-scribble-integrated-sdxl-v2-fp16"],
- "mlsd": ["lllyasviel/control_v11p_sd15_mlsd", "r3gm/controlnet-union-sdxl-1.0-fp16"],
- "scribble": ["lllyasviel/control_v11p_sd15_scribble", "r3gm/controlnet-canny-scribble-integrated-sdxl-v2-fp16"],
- "softedge": ["lllyasviel/control_v11p_sd15_softedge", "r3gm/controlnet-union-sdxl-1.0-fp16"],
- "segmentation": ["lllyasviel/control_v11p_sd15_seg", "r3gm/controlnet-union-sdxl-1.0-fp16"],
- "depth": ["lllyasviel/control_v11f1p_sd15_depth", "r3gm/controlnet-union-sdxl-1.0-fp16"],
- "normalbae": ["lllyasviel/control_v11p_sd15_normalbae", "r3gm/controlnet-union-sdxl-1.0-fp16"],
- "lineart": ["lllyasviel/control_v11p_sd15_lineart", "r3gm/controlnet-union-sdxl-1.0-fp16"],
- "lineart_anime": ["lllyasviel/control_v11p_sd15s2_lineart_anime", "r3gm/controlnet-lineart-anime-sdxl-fp16"],