File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -2189,7 +2189,7 @@ def set_vocab(self):
2189
2189
toktype = SentencePieceTokenTypes .BYTE
2190
2190
# take care of ununsed raw token
2191
2191
if piece .startswith ('[UNUSED' ):
2192
- toktype = SentencePieceTokenTypes .UNKNOWN
2192
+ toktype = SentencePieceTokenTypes .UNUSED
2193
2193
2194
2194
tokens .append (text )
2195
2195
scores .append (score )
@@ -2219,7 +2219,7 @@ def set_vocab(self):
2219
2219
if token == chat_eos_token :
2220
2220
chat_eos_token_id = token_id
2221
2221
token = token .encode ("utf-8" )
2222
- if toktypes [token_id ] != SentencePieceTokenTypes .UNKNOWN :
2222
+ if toktypes [token_id ] != SentencePieceTokenTypes .UNUSED :
2223
2223
assert (tokens [token_id ] == token )
2224
2224
tokens [token_id ] = token
2225
2225
scores [token_id ] = - 1000.0
@@ -2238,7 +2238,7 @@ def set_vocab(self):
2238
2238
if token == chat_eos_token :
2239
2239
chat_eos_token_id = token_id
2240
2240
token = token .encode ("utf-8" )
2241
- if toktypes [token_id ] != SentencePieceTokenTypes .UNKNOWN :
2241
+ if toktypes [token_id ] != SentencePieceTokenTypes .UNUSED :
2242
2242
assert (tokens [token_id ] == token )
2243
2243
tokens [token_id ] = token
2244
2244
scores [token_id ] = - 1000.0
You can’t perform that action at this time.
0 commit comments