From 31c093c2ccc9b602755817569750bfe196a446f4 Mon Sep 17 00:00:00 2001
From: xaedes <xaedes@gmail.com>
Date: Mon, 28 Aug 2023 18:33:00 +0200
Subject: [PATCH] bug fixes for convert-train-checkpoint-to-gguf.py loading
 checkpoints with opt_version=0

---
 .../convert-train-checkpoint-to-gguf.py                     | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/examples/train-text-from-scratch/convert-train-checkpoint-to-gguf.py b/examples/train-text-from-scratch/convert-train-checkpoint-to-gguf.py
index a69a9687d..03e6d6245 100644
--- a/examples/train-text-from-scratch/convert-train-checkpoint-to-gguf.py
+++ b/examples/train-text-from-scratch/convert-train-checkpoint-to-gguf.py
@@ -116,8 +116,8 @@ class OptimizationParamsV0:
         self.n_threads            = struct.unpack('<i', bytes(data[offset:offset + 4]))[0];  offset += 4
         self.past                 = struct.unpack('<i', bytes(data[offset:offset + 4]))[0];  offset += 4
         self.delta                = struct.unpack('<f', bytes(data[offset:offset + 4]))[0];  offset += 4
-        self.print_forward_graph  = struct.unpack('<?', bytes(data[offset:offset + 4]))[0];  offset += 4 # 32bit-aligned
-        self.print_backward_graph = struct.unpack('<?', bytes(data[offset:offset + 4]))[0];  offset += 4 # 32bit-aligned
+        self.print_forward_graph  = struct.unpack('<?', bytes(data[offset:offset + 1]))[0];  offset += 4 # 32bit-aligned
+        self.print_backward_graph = struct.unpack('<?', bytes(data[offset:offset + 1]))[0];  offset += 4 # 32bit-aligned
         self.adam_n_iter          = struct.unpack('<i', bytes(data[offset:offset + 4]))[0];  offset += 4
         self.adam_sched           = struct.unpack('<f', bytes(data[offset:offset + 4]))[0];  offset += 4
         self.adam_decay           = struct.unpack('<f', bytes(data[offset:offset + 4]))[0];  offset += 4
@@ -177,7 +177,7 @@ class OptimizationContext:
                 g  = Tensor('f', [self.nx])
                 g2 = Tensor('f', [self.nx])
                 mh = Tensor('f', [self.nx])
-                mv = Tensor('f', [self.nx])
+                vh = Tensor('f', [self.nx])
 
                 offset = x.load(data, offset)
                 offset = g.load(data, offset)