update

gradio_demo/app.py

@@ -45,14 +45,18 @@ hf_hub_download(repo_id="InstantX/InstantIR", filename="models/previewer_lora_we
 
 instantir_path = f'./models'
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
 sdxl_repo_id = "stabilityai/stable-diffusion-xl-base-1.0"
 dinov2_repo_id = "facebook/dinov2-large"
 lcm_repo_id = "latent-consistency/lcm-lora-sdxl"
 
 if torch.cuda.is_available():
+    device = "cuda"
     torch_dtype = torch.float16
+elif torch.backends.mps.is_available():
+    device = "mps"
+    torch_dtype = torch.float32
 else:
+    device = "cpu"
     torch_dtype = torch.float32
 
 # Load pretrained models.
@@ -262,4 +266,4 @@ with gr.Blocks() as demo:
     ```
     """)
 
-demo.queue().launch()
+demo.queue().launch()