update

app.py

@@ -23,7 +23,13 @@ import gradio as gr
 
 # global variable
 MAX_SEED = np.iinfo(np.int32).max
-device = "cuda" if torch.cuda.is_available() else "cpu"
+#device = "cuda" if torch.cuda.is_available() else "cpu"
+if torch.backends.mps.is_available():
+  device = "mps"
+elif torch.cuda.is_available():
+  device = "cuda"
+else:
+  device = "cpu"
 STYLE_NAMES = list(styles.keys())
 DEFAULT_STYLE_NAME = "Watercolor"
 
@@ -53,10 +59,16 @@ pipe = StableDiffusionXLInstantIDPipeline.from_pretrained(
     safety_checker=None,
     feature_extractor=None,
 )
-pipe.cuda()
+if device == 'mps':
+  pipe.to("mps")
+elif device == 'cuda':
+  pipe.cuda()
 pipe.load_ip_adapter_instantid(face_adapter)
-pipe.image_proj_model.to('cuda')
-pipe.unet.to('cuda')
+#pipe.image_proj_model.to('cuda')
+#pipe.unet.to('cuda')
+if device == 'mps' or device == 'cuda':
+  pipe.image_proj_model.to(device)
+  pipe.unet.to(device)
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed:
@@ -407,4 +419,4 @@ with gr.Blocks(css=css) as demo:
     
     gr.Markdown(article)
 
-demo.launch()
+demo.launch()