change base model

app_haircolor_inpaint_15.py

@@ -24,8 +24,8 @@ from controlnet_aux import (
     HEDdetector,
 )
 
-# BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5"
-BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-inpainting"
+BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5"
+# BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-inpainting"
 # BASE_MODEL = "SG161222/Realistic_Vision_V5.1_noVAE"
 
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
@@ -59,7 +59,8 @@ controlnet = [
 basepipeline = StableDiffusionControlNetInpaintPipeline.from_pretrained(
     BASE_MODEL,
     torch_dtype=torch.float16,
-    use_safetensors=True,
+    # use_safetensors=True,
+    revision="fp16",
     controlnet=controlnet,
 )
 # basepipeline.scheduler = DDIMScheduler.from_config(basepipeline.scheduler.config)