From ba1be979ebc25f24c167e77a16840b21ec12c926 Mon Sep 17 00:00:00 2001
From: Christian Zhou-Zheng <christianzhouzheng@gmail.com>
Date: Sun, 9 Jun 2024 11:10:33 -0400
Subject: [PATCH] fix ti data messiness

---
 gguf-py/gguf/gguf_writer.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/gguf-py/gguf/gguf_writer.py b/gguf-py/gguf/gguf_writer.py
index 0a8749d7f..2b64c6c0d 100644
--- a/gguf-py/gguf/gguf_writer.py
+++ b/gguf-py/gguf/gguf_writer.py
@@ -194,7 +194,8 @@ class GGUFWriter:
         
         total_tensors = sum(len(t) for t in self.tensors)
         for i in range(len(self.fout)):
-            try: # TODO better way to do this
+            # just see whether it exists
+            try:
                 self.kv_data[i]
             except IndexError:
                 self.kv_data.append(dict())
@@ -217,7 +218,6 @@ class GGUFWriter:
 
         for i in range(len(self.fout)):
             fout = self.fout[i]
-            #print(f"writing header: GGUF_VERSION={GGUF_VERSION}, GGUF_MAGIC={GGUF_MAGIC}, n_tensors={len(self.tensors[i])}, n_kv_data={len(self.kv_data[i])}")
             self._write_packed(fout, "<I", GGUF_MAGIC, skip_pack_prefix = True)
             self._write_packed(fout, "I", GGUF_VERSION)
             self._write_packed(fout, "Q", len(self.tensors[i]))
@@ -256,8 +256,8 @@ class GGUFWriter:
                 ti_data += self._pack_val(name, GGUFValueType.STRING, add_vtype=False)
                 n_dims = len(ti.shape)
                 ti_data += self._pack("I", n_dims)
-                for i in range(n_dims):
-                    ti_data += self._pack("Q", ti.shape[n_dims - 1 - i])
+                for j in range(n_dims):
+                    ti_data += self._pack("Q", ti.shape[n_dims - 1 - j])
                 ti_data += self._pack("I", ti.dtype)
                 ti_data += self._pack("Q", offset_tensor)
                 offset_tensor += GGUFWriter.ggml_pad(ti.nbytes, self.data_alignment)