change base model

app_haircolor_inpaint_15.py

@@ -13,6 +13,7 @@ from diffusers import (
     StableDiffusionControlNetInpaintPipeline,
     ControlNetModel,
     DDIMScheduler,
+    DPMSolverMultistepScheduler,
 )
 
 from controlnet_aux import (
@@ -22,7 +23,8 @@ from controlnet_aux import (
     HEDdetector,
 )
 
-BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5"
+# BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5"
+BASE_MODEL = "SG161222/Realistic_Vision_V2.0"
 
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -58,7 +60,11 @@ basepipeline = StableDiffusionControlNetInpaintPipeline.from_pretrained(
     use_safetensors=True,
     controlnet=controlnet,
 )
-basepipeline.scheduler = DDIMScheduler.from_config(basepipeline.scheduler.config)
+# basepipeline.scheduler = DDIMScheduler.from_config(basepipeline.scheduler.config)
+basepipeline.scheduler = DPMSolverMultistepScheduler.from_config(
+    basepipeline.scheduler.config,
+    use_karras_sigmas=True,
+)
 
 basepipeline = basepipeline.to(DEVICE)