Update app.py

app.py

@@ -31,8 +31,13 @@ with open(lora_list, "r") as file:
 
 for item in sdxl_loras:
     saved_name = hf_hub_download(item["repo"], item["weights"])
+    
+    if saved_name.endswith('.safetensors'):
+        state_dict = load_file(saved_name)
+    else:
+        state_dict = torch.load(saved_name)
+        
     item["saved_name"] = saved_name
-    state_dict = load_file(saved_name)
     item["state_dict"] = {k: v.to(device="cuda", dtype=torch.float16) for k, v in state_dict.items() if torch.is_tensor(v)}
 
 css = '''