From 5d22a9db133d0b5737f8889d488070a55dbe7682 Mon Sep 17 00:00:00 2001
From: klosax <131523366+klosax@users.noreply.github.com>
Date: Mon, 14 Aug 2023 10:55:44 +0200
Subject: [PATCH] convert-gptneox-h5-to-gguf.py : tensor name map changes

---
 convert-gptneox-h5-to-gguf.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/convert-gptneox-h5-to-gguf.py b/convert-gptneox-h5-to-gguf.py
index ba6e90e42..a3d78c9ef 100644
--- a/convert-gptneox-h5-to-gguf.py
+++ b/convert-gptneox-h5-to-gguf.py
@@ -1,7 +1,7 @@
 # HF gptneox--> gguf conversion
 
 import gguf
-import gguf_tensor_map as tmap
+import gguf_namemap as tmap
 import os
 import sys
 import struct
@@ -188,7 +188,7 @@ if Path(dir_model + "/tokenizer.json").is_file():
 
 # TENSORS
 
-tensor_map = tmap.get_tensor_map(block_count)
+tensor_map = tmap.get_tensor_namemap(block_count)
 
 # tensor info
 print("gguf: get tensor metadata")