convert-gptneox-h5-to-gguf.py : tensor name map changes

This commit is contained in:
klosax
2023-08-14 10:55:44 +02:00
committed by GitHub
parent 56a1f32072
commit 5d22a9db13

View File

@@ -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")