update

app.py

@@ -30,7 +30,7 @@ elif torch.backends.mps.is_available():
 else:
     device = torch.device("cpu")
 #device = "cuda" if torch.cuda.is_available() else "cpu"
-dtype = torch.float16 if str(device).__contains__("cuda") else torch.float32
+dtype = torch.float16 if str(device).__contains__("cuda") or str(device).__contains__("mps") else torch.float32
 
 # initialization
 base_model_path = "stabilityai/stable-diffusion-xl-base-1.0"
@@ -38,13 +38,13 @@ image_encoder_path = "sdxl_models/image_encoder"
 ip_ckpt = "sdxl_models/ip-adapter_sdxl.bin"
 
 controlnet_path = "diffusers/controlnet-canny-sdxl-1.0"
-controlnet = ControlNetModel.from_pretrained(controlnet_path, use_safetensors=False, torch_dtype=torch.float16).to(device)
+controlnet = ControlNetModel.from_pretrained(controlnet_path, use_safetensors=False, torch_dtype=dtype).to(device)
 
 # load SDXL pipeline
 pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
     base_model_path,
     controlnet=controlnet,
-    torch_dtype=torch.float16,
+    torch_dtype=dtype,
     add_watermarker=False,
 )