Compare commits

...
Sign in to create a new pull request.

2 commits

Author SHA1 Message Date
Brian
284870c868
Merge branch 'master' into fix-convert-modelname 2024-05-14 15:05:49 +10:00
Jared Van Bortel
941de11759 convert : get general.name from model dir, not its parent 2024-02-20 11:17:16 -05:00

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"