Narsil HF staff commited on
Commit
d63c3cd
1 Parent(s): cee67d6

Update convert.py

Browse files
Files changed (1) hide show
  1. convert.py +1 -1
convert.py CHANGED
@@ -183,7 +183,7 @@ def convert_file(
183
  sf_filename: str,
184
  discard_names: List[str],
185
  ):
186
- loaded = torch.load(pt_filename, map_location="cpu")
187
  if "state_dict" in loaded:
188
  loaded = loaded["state_dict"]
189
  to_removes = _remove_duplicate_names(loaded, discard_names=discard_names)
 
183
  sf_filename: str,
184
  discard_names: List[str],
185
  ):
186
+ loaded = torch.load(pt_filename, map_location="cpu", weights_only=True)
187
  if "state_dict" in loaded:
188
  loaded = loaded["state_dict"]
189
  to_removes = _remove_duplicate_names(loaded, discard_names=discard_names)