File tree Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change 22from abc import ABC , abstractmethod
33from typing import Optional
44
5+ import tokenizers
6+ from packaging import version
57from tokenizers import Tokenizer
68from tokenizers .decoders import DecodeStream
79from transformers import PreTrainedTokenizerFast
@@ -43,8 +45,10 @@ def from_new_request(
4345 # No tokenizer => skipping detokenization.
4446 return IncrementalDetokenizer ()
4547
46- if isinstance (tokenizer , PreTrainedTokenizerFast ):
48+ if (isinstance (tokenizer , PreTrainedTokenizerFast ) and version .parse (
49+ tokenizers .__version__ ) >= version .parse ("0.21.1" )):
4750 # Fast tokenizer => use tokenizers library DecodeStream.
51+ # And only tokenizers >= 0.21.1 supports Fast Detokenizer.
4852 return FastIncrementalDetokenizer (tokenizer , request )
4953
5054 # Fall back to slow python-based incremental detokenization.
You can’t perform that action at this time.
0 commit comments