From d7003a98cceda5fe5926baf3dcad666a311dbe40 Mon Sep 17 00:00:00 2001
From: xaedes <xaedes@gmail.com>
Date: Sat, 17 Jun 2023 18:56:27 +0200
Subject: [PATCH] Fix reset of unused g->nodes and g->grads to NULL

---
 examples/train-text-from-scratch/train-text-from-scratch.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/train-text-from-scratch/train-text-from-scratch.cpp b/examples/train-text-from-scratch/train-text-from-scratch.cpp
index a4a6b05b1..c76a80c75 100644
--- a/examples/train-text-from-scratch/train-text-from-scratch.cpp
+++ b/examples/train-text-from-scratch/train-text-from-scratch.cpp
@@ -1403,8 +1403,8 @@ void graph_set_leafs_grads(struct ggml_cgraph * g) {
         }
     }
     for (int i=n_nodes; i < g->n_nodes; ++i) {
-        g->nodes[n_nodes] = NULL;
-        g->grads[n_nodes] = NULL;
+        g->nodes[i] = NULL;
+        g->grads[i] = NULL;
     }
     g->n_nodes = n_nodes;
 }