From ce865b3ce3a61cfa6277bb152f053d03345c1980 Mon Sep 17 00:00:00 2001 From: KerfuffleV2 Date: Tue, 7 Nov 2023 18:43:23 -0700 Subject: [PATCH] Fix missing return statement in add_tensor --- gguf-py/gguf/gguf_writer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/gguf-py/gguf/gguf_writer.py b/gguf-py/gguf/gguf_writer.py index bf7c4a7df..99d4d70a8 100644 --- a/gguf-py/gguf/gguf_writer.py +++ b/gguf-py/gguf/gguf_writer.py @@ -228,6 +228,7 @@ class GGUFWriter: if self.temp_file is None: self.tensors.append(tensor) + return tensor.tofile(self.temp_file) self.write_padding(self.temp_file, tensor.nbytes)