summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJared Van Bortel <jared@nomic.ai>2024-05-16 02:15:23 -0400
committerGitHub <noreply@github.com>2024-05-16 16:15:23 +1000
commitdda64fc17c97820ea9489eb0cc9ae8b8fdce4926 (patch)
tree1840d079f6d5b584e5b5453cc393201850da3c8a
parent0350f5815218c483fb3026a86adc44a115481625 (diff)
convert : get general.name from model dir, not its parent (#5615)
Co-authored-by: Brian <mofosyne@gmail.com>
-rwxr-xr-xconvert.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/convert.py b/convert.py
index e2e64235..da124795 100755
--- a/convert.py
+++ b/convert.py
@@ -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"