Update convert_safetensors.py

solve bug Signed-off-by: yuunnn_w <[email protected]>
Ai00-X · Feb 27, 2024 · 153c9c4 · 153c9c4
1 parent 7e7984f
commit 153c9c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/convert_safetensors.py b/convert_safetensors.py
@@ -65,7 +65,7 @@ def convert_file(pt_filename: str, sf_filename: str, rename={}, transpose_names=
         loaded[new_k] = {
             "dtype": str(v.dtype).split(".")[-1],
             "shape": v.shape,
-            "data": v.numpy().tobytes(),
+            "data": v.detach().numpy().tobytes(),
         }
 
     dirname = os.path.dirname(sf_filename)