update

app.py

@@ -59,7 +59,13 @@ with open("defaults_data.json", "r") as file:
     lora_defaults = json.load(file)
     
 
-device = "cuda" 
+#device = "cuda" 
+if torch.cuda.is_available():
+    device = "cuda"
+elif torch.backends.mps.is_available():
+    device = "mps"
+else:
+    device = "cpu"
 
 state_dicts = {}
 
@@ -468,4 +474,4 @@ with gr.Blocks(css="custom.css") as demo:
     share_button.click(None, [], [], js=share_js)
     demo.load(fn=classify_gallery, inputs=[gr_sdxl_loras], outputs=[gallery, gr_sdxl_loras], queue=False, js=js)
 demo.queue(max_size=20)
-demo.launch(share=True)
+demo.launch(share=True)

requirements.txt

@@ -9,4 +9,5 @@ gdown
 onnxruntime-gpu
 peft
 compel
-gradio_imageslider
+gradio_imageslider
+gradio