vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

convert : get general.name from model dir, not its parent

Browse source
This commit is contained in:
Jared Van Bortel 2024-02-20 11:16:54 -05:00
parent f24ed14ee0
commit 941de11759
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
convert.py
View file

@ -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)