change to 1.5

app_haircolor.py

@@ -9,6 +9,7 @@ from segment_utils import(
     restore_result,
 )
 from diffusers import (
+    StableDiffusionAdapterPipeline,
     DiffusionPipeline,
     T2IAdapter,
     MultiAdapter,
@@ -21,7 +22,7 @@ from controlnet_aux import (
     MidasDetector,
 )
 
-BASE_MODEL = "stabilityai/sdxl-turbo"
+BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
 DEFAULT_EDIT_PROMPT = "blue hair"
@@ -59,12 +60,11 @@ adapters = MultiAdapter(
 adapters = adapters.to(torch.float16)
 
 
-basepipeline = DiffusionPipeline.from_pretrained(
+basepipeline = StableDiffusionAdapterPipeline.from_pretrained(
     BASE_MODEL,
     torch_dtype=torch.float16,
     use_safetensors=True,
     adapter=adapters,
-    custom_pipeline="./pipelines/pipeline_sdxl_adapter_img2img.py",
 )
 
 basepipeline = basepipeline.to(DEVICE)