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

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 teleprint-me
parent d0a9c31e08
commit c7a926f970
No known key found for this signature in database
GPG key ID: B0D11345E65C4D48

View file

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