From 941de11759e3957731ac98331154f1beaad62a62 Mon Sep 17 00:00:00 2001 From: Jared Van Bortel Date: Tue, 20 Feb 2024 11:16:54 -0500 Subject: [PATCH] convert : get general.name from model dir, not its parent --- convert.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/convert.py b/convert.py index 63a0a5d78..63285f238 100755 --- a/convert.py +++ b/convert.py @@ -974,7 +974,7 @@ class OutputFile: if params.n_ctx == 4096: name = "LLaMA v2" elif params.path_model is not None: - name = str(params.path_model.parent).split('/')[-1] + name = params.path_model.name self.gguf.add_name (name) self.gguf.add_context_length (params.n_ctx)