convert

Sleeping

Narsil HF staff commited on Nov 30, 2022

Commit

c968ad4

•

1 Parent(s): cbdc5ec

Update convert.py

Files changed (1) hide show

convert.py CHANGED Viewed

@@ -95,7 +95,7 @@ def convert_file(
     pt_filename: str,
     sf_filename: str,
 ):
-    loaded = torch.load(pt_filename)
     if "state_dict" in loaded:
         loaded = loaded["state_dict"]
     shared = shared_pointers(loaded)

     pt_filename: str,
     sf_filename: str,
 ):
+    loaded = torch.load(pt_filename, map_location="cpu")
     if "state_dict" in loaded:
         loaded = loaded["state_dict"]
     shared = shared_pointers(loaded)