Update convert.py
Browse files- 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)
|