update torch load

app.py

@@ -22,6 +22,7 @@ from PIL import Image
 
 MY_TOKEN = os.environ.get("MY_TOKEN")
 
+print(torch.cuda.is_available())
 
 def image_to_base64(image: Image.Image) -> str:
     buffered = BytesIO()
@@ -77,7 +78,9 @@ for item in sdxl_loras_raw:
     saved_name = hf_hub_download(item["repo"], item["weights"])
 
     if not saved_name.endswith('.safetensors'):
-        state_dict = torch.load(saved_name)
+        # state_dict = torch.load(saved_name)
+        state_dict = torch.load(saved_name, map_location=torch.device('cpu'))
+
     else:
         state_dict = load_file(saved_name)
 
@@ -153,10 +156,10 @@ def run_lora(prompt, negative, lora_scale, selected_state, sdxl_loras, progress=
     global last_lora, last_merged, last_fused, pipe
 
     print("✅ Running LoRAAAAA  >>>>>>>>>>>>> >>>>>>>>>>>>>>>>>>>")
-    print("prompt: ", prompt)
-    print("negative: ", negative)
-    print("lora_scale: ", lora_scale)
-    print("selected_state: ", selected_state)
+    # print("prompt: ", prompt)
+    # print("negative: ", negative)
+    # print("lora_scale: ", lora_scale)
+    # print("selected_state: ", selected_state)
     print("selected_state index: ", selected_state.index)