yozozaya commited on
Commit
ae0ca6f
·
1 Parent(s): eff8fcf

update torch load

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- # 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)
 
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)