diff options
| author | cebtenzzre <cebtenzzre@gmail.com> | 2023-10-02 14:58:46 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-10-02 14:58:46 -0400 |
| commit | 0fe321031a5c670ab5fb5f49d69c4c91d783c93f (patch) | |
| tree | 4894e8f94e88165d37bf5c278fae4540a8a06cd0 /examples/train-text-from-scratch/convert-train-checkpoint-to-gguf.py | |
| parent | 9476b012260a2fb6c67976582d64484ce7406ed9 (diff) | |
gguf : general usability improvements (#3409)
Diffstat (limited to 'examples/train-text-from-scratch/convert-train-checkpoint-to-gguf.py')
| -rw-r--r-- | examples/train-text-from-scratch/convert-train-checkpoint-to-gguf.py | 2 |
1 files changed, 1 insertions, 1 deletions
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): |