From c7bd8c147ccd7411d1cd9a2a394d2d9b743d7294 Mon Sep 17 00:00:00 2001
From: klosax <131523366+klosax@users.noreply.github.com>
Date: Sun, 13 Aug 2023 00:03:32 +0200
Subject: [PATCH] gptneox-main.cpp : n_layer --> n_block

---
 gptneox-main.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gptneox-main.cpp b/gptneox-main.cpp
index 7420daf96..4773dfd69 100644
--- a/gptneox-main.cpp
+++ b/gptneox-main.cpp
@@ -414,7 +414,7 @@ bool gpt_neox_model_load(const std::string & fname, gpt_neox_model & model, gpt2
         if (ok) { keyidx = gguf_find_key(ggufctx, "gptneox.attention.head_count");
                   if (keyidx != -1) { hparams.n_head = gguf_get_val_u32(ggufctx, keyidx); } else { ok = false; }  }
 
-        if (ok) { keyidx = gguf_find_key(ggufctx, "gptneox.layer_count");
+        if (ok) { keyidx = gguf_find_key(ggufctx, "gptneox.block_count");
                   if (keyidx != -1) { hparams.n_block = gguf_get_val_u32(ggufctx, keyidx); } else { ok = false; }  }
 
         if (ok) { keyidx = gguf_find_key(ggufctx, "gptneox.rope.dimension_count");