From bb54d1700e5071c40bc9da4a56314b69f1117b1b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?M=2E=20Yusuf=20Sar=C4=B1g=C3=B6z?= <yusufsarigoz@gmail.com>
Date: Thu, 27 Jul 2023 23:09:53 +0300
Subject: [PATCH] GGUF : Support writing tensors in Python

---
 gguf.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/gguf.py b/gguf.py
index 1f24cb9d5..f1e2c69a2 100644
--- a/gguf.py
+++ b/gguf.py
@@ -62,7 +62,7 @@ class GGUFWriter:
     def __init__(self, fout: IO):
         self.fout = fout
         self.offset_tensor = 0
-        self.tensors = []
+        self.tensors: List[np.ndarray] = []
 
     def write_header(self, tensor_count: int, metadata_kv_count: int):
         self.fout.write(struct.pack("<I", constants.GGUF_MAGIC))
@@ -176,6 +176,12 @@ class GGUFWriter:
 
         self.tensors.append(tensor)
 
+    def write_tensors(self):
+        for tensor in self.tensors:
+            tensor.tofile(self.fout)
+            pad = GGUFWriter.ggml_pad(tensor.nbytes, constants.GGUF_DEFAULT_ALIGNMENT) - tensor.nbytes
+            self.fout.write(bytes([0] * pad))
+
     def flush(self):
         self.fout.flush()