Fix safe open. (#8422)

PaddlePaddle · May 13, 2024 · fe6277f · fe6277f
1 parent 6bd2b0d
commit fe6277f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddlenlp/utils/safetensors.py b/paddlenlp/utils/safetensors.py
@@ -273,7 +273,7 @@ def nbytes(self):
 
 # a simple file writer object
 class fast_safe_open:
-    def __init__(self, filename, framework=None):
+    def __init__(self, filename, framework=None, device="cpu"):
         self.filename = filename
         self.framework = framework
         self.file = open(self.filename, "rb")