We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 066b653 commit f0bd60aCopy full SHA for f0bd60a
tensorrt_llm/tokenizer/tokenizer.py
@@ -213,6 +213,13 @@ def trtllm_decode_incrementally(
213
214
new_tokens = self.convert_ids_to_tokens(
215
token_ids, skip_special_tokens=skip_special_tokens)
216
+ # filter out None tokens
217
+ if None in new_tokens:
218
+ logger.warning(
219
+ "An unexpected \"None\" token was generated. This may be caused by a generated token ID being out of the "
220
+ "tokenizer's vocabulary. Filtering out \"None\" tokens from the newly generated tokens."
221
+ )
222
+ new_tokens = [token for token in new_tokens if token is not None]
223
pending_tokens.extend(new_tokens)
224
225
curr_new_text = self.convert_tokens_to_string(
0 commit comments