diff --git a/convert.py b/convert.py index c3f3fc0a1..c6601eaca 100755 --- a/convert.py +++ b/convert.py @@ -873,7 +873,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)