]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
convert : get general.name from model dir, not its parent (#5615)
authorJared Van Bortel <redacted>
Thu, 16 May 2024 06:15:23 +0000 (02:15 -0400)
committerGitHub <redacted>
Thu, 16 May 2024 06:15:23 +0000 (16:15 +1000)
Co-authored-by: Brian <redacted>
convert.py

index e2e6423516d15fd113c1b94db70e9637e012bafe..da1247957780c725147e6f6e7bfd961260d5fc7e 100755 (executable)
@@ -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"