update

app.py

@@ -23,6 +23,9 @@ MAX_SEED = np.iinfo(np.int32).max
 if torch.cuda.is_available():
     device = "cuda"
     dtype = torch.float16
+elif torch.backends.mps.is_available():
+    device = "mps"
+    #dtype = torch.float32
 else:
     device = "cpu"
     dtype = torch.float32
@@ -226,4 +229,4 @@ with gr.Blocks(css=css) as demo:
     
     gr.Markdown(Footer)
 
-demo.launch()
+demo.launch()