Просмотр исходного кода

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

Co-authored-by: Brian <mofosyne@gmail.com>
Jared Van Bortel 1 год назад
Родитель
Сommit
dda64fc17c
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      convert.py

+ 1 - 1
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"