param.path_model is a Path() not a string

This commit is contained in:
brian khuu 2024-01-10 10:46:41 +11:00
parent 8c58330318
commit c410ab8fd7

View file

@ -873,7 +873,7 @@ class OutputFile:
if params.n_ctx == 4096: if params.n_ctx == 4096:
name = "LLaMA v2" name = "LLaMA v2"
elif params.path_model is not None: 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_name (name)
self.gguf.add_context_length (params.n_ctx) self.gguf.add_context_length (params.n_ctx)