update torch load
Browse files
app.py
CHANGED
@@ -78,8 +78,8 @@ for item in sdxl_loras_raw:
|
|
78 |
saved_name = hf_hub_download(item["repo"], item["weights"])
|
79 |
|
80 |
if not saved_name.endswith('.safetensors'):
|
81 |
-
|
82 |
-
state_dict = torch.load(saved_name, map_location=torch.device('cpu'))
|
83 |
|
84 |
else:
|
85 |
state_dict = load_file(saved_name)
|
|
|
78 |
saved_name = hf_hub_download(item["repo"], item["weights"])
|
79 |
|
80 |
if not saved_name.endswith('.safetensors'):
|
81 |
+
state_dict = torch.load(saved_name)
|
82 |
+
# state_dict = torch.load(saved_name, map_location=torch.device('cpu'))
|
83 |
|
84 |
else:
|
85 |
state_dict = load_file(saved_name)
|