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/finetune/convert-finetune-checkpoint-to-gguf.py | |
| parent | 9476b012260a2fb6c67976582d64484ce7406ed9 (diff) | |
gguf : general usability improvements (#3409)
Diffstat (limited to 'examples/finetune/convert-finetune-checkpoint-to-gguf.py')
| -rw-r--r-- | examples/finetune/convert-finetune-checkpoint-to-gguf.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/finetune/convert-finetune-checkpoint-to-gguf.py b/examples/finetune/convert-finetune-checkpoint-to-gguf.py index 96d6633e..c8e14da8 100644 --- a/examples/finetune/convert-finetune-checkpoint-to-gguf.py +++ b/examples/finetune/convert-finetune-checkpoint-to-gguf.py @@ -313,7 +313,7 @@ class ModelParams: gguf_writer.add_feed_forward_length(self.get_n_ff()) def tensor_name(key, bid=None, suffix=".weight"): - return gguf.MODEL_TENSOR_NAMES[gguf.MODEL_ARCH.LLAMA][key].format(bid=bid) + suffix + return gguf.TENSOR_NAMES[key].format(bid=bid) + suffix class Layer: def __init__(self, params, lora_params, bid): |