tristan/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 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"