convert : get general.name from model dir, not its parent
This commit is contained in:
parent
f24ed14ee0
commit
941de11759
|
|
@ -974,7 +974,7 @@ class OutputFile:
|
|||
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)
|
||||
|
|
|
|||
Loading…
Reference in New Issue