update

app.py

@@ -5,23 +5,25 @@ from SDXL.diff_pipe import StableDiffusionXLDiffImg2ImgPipeline
 from diffusers import DPMSolverMultistepScheduler
 
 NUM_INFERENCE_STEPS = 50
+dtype = torch.float16
 if torch.cuda.is_available():
   device = "cuda"
 elif torch.backends.mps.is_available():
   device = "mps"
+  dtype = torch.float32
 else:
   device = "cpu"
 #device = "cuda"
 
 base = StableDiffusionXLDiffImg2ImgPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
+    "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=dtype, variant="fp16", use_safetensors=True
 )
 
 refiner = StableDiffusionXLDiffImg2ImgPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-refiner-1.0",
     text_encoder_2=base.text_encoder_2,
     vae=base.vae,
-    torch_dtype=torch.float16,
+    torch_dtype=dtype,
     use_safetensors=True,
     variant="fp16",
 )