convert : get general.name from model dir, not its parent (#5615)
Co-authored-by: Brian <mofosyne@gmail.com>
This commit is contained in:
parent
0350f58152
commit
dda64fc17c
1 changed files with 1 additions and 1 deletions
|
@ -1109,7 +1109,7 @@ class OutputFile:
|
|||
if metadata is not None and metadata.name is not None:
|
||||
name = metadata.name
|
||||
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:
|
||||
# Heuristic detection of LLaMA v2 model
|
||||
name = "LLaMA v2"
|
||||
|
|
Loading…
Add table
Reference in a new issue