From 054776049ebe6fcab96b2bc6091a7387c9c4a6f4 Mon Sep 17 00:00:00 2001 From: KerfuffleV2 Date: Mon, 21 Aug 2023 08:27:31 -0600 Subject: [PATCH] Set default value for gguf add_tensor raw_shape KW arg --- gguf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gguf.py b/gguf.py index 5c82414fa..9dbde2d54 100644 --- a/gguf.py +++ b/gguf.py @@ -509,7 +509,7 @@ class GGUFWriter: self.offset_tensor += GGUFWriter.ggml_pad(tensor_nbytes, self.data_alignment) self.ti_data_count += 1 - def add_tensor(self, name: str, tensor: np.ndarray, raw_shape: Optional[np.ndarray], raw_dtype: Optional[GGMLQuantizationType] = None): + def add_tensor(self, name: str, tensor: np.ndarray, raw_shape: Optional[np.ndarray] = None, raw_dtype: Optional[GGMLQuantizationType] = None): if self.use_temp_file and not hasattr(self, "temp_file"): self.temp_file = tempfile.SpooledTemporaryFile(mode="w+b", max_size=256*1024*1024) self.temp_file.seek(0)