Skip to content

Commit 7febaf0

Browse files
add qwen2tokenizer to fast
1 parent f42cc42 commit 7febaf0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

optimum/intel/openvino/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@
3131
from openvino import Core, Model, properties
3232
from openvino import Type as OVType
3333
from packaging.version import Version
34-
from transformers import AutoTokenizer, CLIPTokenizer, PreTrainedTokenizer, PreTrainedTokenizerFast
34+
from transformers import AutoTokenizer, CLIPTokenizer, Qwen2Tokenizer, PreTrainedTokenizer, PreTrainedTokenizerFast
3535
from transformers.onnx.utils import ParameterFormat, compute_serialized_parameters_size
3636

3737
from optimum.intel.utils.import_utils import is_torch_version
@@ -206,7 +206,7 @@
206206
}
207207

208208

209-
NEED_CONVERT_TO_FAST_TOKENIZER: Tuple[Type[PreTrainedTokenizer]] = (CLIPTokenizer,)
209+
NEED_CONVERT_TO_FAST_TOKENIZER: Tuple[Type[PreTrainedTokenizer]] = (CLIPTokenizer, Qwen2Tokenizer)
210210

211211

212212
def maybe_convert_tokenizer_to_fast(

0 commit comments

Comments
 (0)