Compare commits

...

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"