root/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/llama.cpp.git synced 2024-12-25 02:44:36 +00:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

convert : get general.name from model dir, not its parent (#5615)

Browse Source 
Co-authored-by: Brian <mofosyne@gmail.com>
This commit is contained in:
Jared Van Bortel 2024-05-16 02:15:23 -04:00 committed by GitHub
parent 0350f58152
commit dda64fc17c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
convert.py
View File

@ -1109,7 +1109,7 @@ class OutputFile:
if metadata is not None and metadata.name is not None: if metadata is not None and metadata.name is not None:
name = metadata.name name = metadata.name
elif params.path_model is not None: elif params.path_model is not None:
name = str(params.path_model.parent).split("/")[-1] name = params.path_model.name
elif params.n_ctx == 4096: elif params.n_ctx == 4096:
# Heuristic detection of LLaMA v2 model # Heuristic detection of LLaMA v2 model
name = "LLaMA v2" name = "LLaMA v2"