Spaces:

safetensors
/

convert

Running

Narsil HF staff commited on Nov 26, 2022

Commit

ba06fbc

•

1 Parent(s): bac319c

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)
     shared = shared_pointers(loaded)
     for shared_weights in shared:
         for name in shared_weights[1:]:

     pt_filename: str,
     sf_filename: str,
 ):
+    loaded = torch.load(pt_filename, map_location="cpu")
     shared = shared_pointers(loaded)
     for shared_weights in shared:
         for name in shared_weights[1:]: