param.path_model is a Path() not a string

ggerganov · Jan 9, 2024 · c410ab8 · c410ab8
1 parent 8c58330
commit c410ab8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/convert.py b/convert.py
@@ -873,7 +873,7 @@ def add_meta_arch(self, params: Params) -> None:
         if params.n_ctx == 4096:
             name = "LLaMA v2"
         elif params.path_model is not None:
-            name = str(params.path_model.parent).split('/')[-1]
+            name = params.path_model.name
 
         self.gguf.add_name                (name)
         self.gguf.add_context_length      (params.n_ctx)