Spaces:

multimodalart
/

lora-roulette

Runtime error

multimodalart HF staff commited on Sep 20, 2023

Commit

4aa6570

•

1 Parent(s): 71d0bce

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -51,7 +51,7 @@ for item in sdxl_loras:
     saved_name = hf_hub_download(item["repo"], item["weights"])
     if not saved_name.endswith('.safetensors'):
-        state_dict = torch.load(saved_name)
     else:
         state_dict = load_file(saved_name)

     saved_name = hf_hub_download(item["repo"], item["weights"])
     if not saved_name.endswith('.safetensors'):
+        state_dict = torch.load(saved_name, map_location=torch.device('cpu'))
     else:
         state_dict = load_file(saved_name)