convert : revert "added_tokens_decoder" change

This commit is contained in:
Georgi Gerganov 2023-12-16 16:05:35 +02:00
parent a878be4cb1
commit 0b6ffa580c
No known key found for this signature in database
GPG key ID: 449E073F9DC10735

View file

@ -249,7 +249,7 @@ class Model:
toktypes.append(gguf.TokenType.USER_DEFINED)
elif reverse_vocab[i] in added_vocab:
tokens.append(reverse_vocab[i])
if hasattr(tokenizer, "added_tokens_decoder") and tokenizer.added_tokens_decoder[i].special:
if tokenizer.added_tokens_decoder[i].special:
toktypes.append(gguf.TokenType.CONTROL)
else:
toktypes.append(gguf.TokenType.USER_DEFINED)