From 0fe321031a5c670ab5fb5f49d69c4c91d783c93f Mon Sep 17 00:00:00 2001 From: cebtenzzre Date: Mon, 2 Oct 2023 14:58:46 -0400 Subject: gguf : general usability improvements (#3409) --- examples/train-text-from-scratch/convert-train-checkpoint-to-gguf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'examples/train-text-from-scratch/convert-train-checkpoint-to-gguf.py') 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 351e7bc2..887ed2e2 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 @@ -364,7 +364,7 @@ class ModelParams: gguf_writer.add_feed_forward_length(self.get_n_ff()) def tensor_name(key, bid=None): - return gguf.MODEL_TENSOR_NAMES[gguf.MODEL_ARCH.LLAMA][key].format(bid=bid) + ".weight" + return gguf.TENSOR_NAMES[key].format(bid=bid) + ".weight" class Layer: def __init__(self, params, bid): -- cgit v1.2.3