From 0c219fb5b56d05d54f10811826a768fe039136e2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?M=2E=20Yusuf=20Sar=C4=B1g=C3=B6z?= <yusufsarigoz@gmail.com>
Date: Sat, 29 Jul 2023 12:42:54 +0300
Subject: [PATCH] gguf : fix writing gguf arrays

---
 gguf.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/gguf.py b/gguf.py
index c5a8f5b90..de75b54d4 100644
--- a/gguf.py
+++ b/gguf.py
@@ -147,9 +147,10 @@ class GGUFWriter:
             self.fout.write(struct.pack("<I", len(encoded_val)))
             self.fout.write(encoded_val)
         elif vtype == GGUFValueType.ARRAY:
+            ltype = set([GGUFValueType.get_type(item) for item in val])
+            assert len(ltype) == 1, "All items in a GGUF array should be of the same type"
+            self.fout.write(struct.pack("<I", ltype[0]))
             self.fout.write(struct.pack("<I", len(val)))
-            # TODO: verify that all elements are of the same type
-            self.fout.write(struct.pack("<I", GGUFValueType.get_type(val[0])))
             for item in val:
                 self.write_val(item, write_vtype=False)
         else: