convert.py : use dir name to name the llama

This commit is contained in:
Georgi Gerganov 2023-08-30 13:29:40 +03:00
parent c90d135eb4
commit b532a69b2f
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735

View File

@ -811,9 +811,11 @@ class OutputFile:
def add_meta_arch(self, params: Params) -> None:
name = "LLaMA"
# TODO: better logic to determine model name
if (params.n_ctx == 4096):
name = "LLaMA v2"
if params.path_model:
elif params.path_model:
name = str(params.path_model.parent).split('/')[-1]
self.gguf.add_name (name)
@ -839,8 +841,7 @@ class OutputFile:
tokens = []
scores = []
toktypes = []
# NOTE: `all_tokens` returns the the base vocabulary and added tokens
# TODO: add special tokens?
# NOTE: `all_tokens` returns the base vocabulary and added tokens
for text, score, toktype in vocab.all_tokens():
tokens.append(text)
scores.append(score)