Skip to content

Commit 96ea0a3

Browse files
committed
Add OV Tokenizers Version Check
1 parent dd31f88 commit 96ea0a3

File tree

2 files changed

+14
-5
lines changed

2 files changed

+14
-5
lines changed

optimum/exporters/openvino/convert.py

+7-2
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@
4040
_timm_version,
4141
_torch_version,
4242
_transformers_version,
43+
compare_versions,
4344
)
4445
from optimum.utils import DEFAULT_DUMMY_SHAPES, is_diffusers_available
4546
from optimum.utils.save_utils import maybe_save_preprocessors
@@ -723,8 +724,12 @@ def export_tokenizer(
723724
if output.exists():
724725
tokenizer = maybe_convert_tokenizer_to_fast(tokenizer, output)
725726

726-
if task is not None and task.startswith("text-generation"):
727-
logger.info(f"Set padding side to left for `{task}` task.")
727+
if (
728+
task is not None
729+
and task.startswith("text-generation")
730+
and compare_versions("openvino-tokenizers", ">=", "2024.3.0.0")
731+
):
732+
logger.info(f"Set tokenizer padding side to left for `{task}` task.")
728733
tokenizer.padding_side = "left"
729734
tokenizer.truncation_side = "left"
730735

tests/openvino/test_exporters_cli.py

+7-3
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,11 @@
4141
)
4242
from optimum.intel.openvino.configuration import _DEFAULT_4BIT_CONFIGS
4343
from optimum.intel.openvino.utils import _HEAD_TO_AUTOMODELS
44-
from optimum.intel.utils.import_utils import is_openvino_tokenizers_available, is_transformers_version
44+
from optimum.intel.utils.import_utils import (
45+
compare_versions,
46+
is_openvino_tokenizers_available,
47+
is_transformers_version,
48+
)
4549

4650

4751
class OVCLIExportTestCase(unittest.TestCase):
@@ -171,8 +175,8 @@ def test_exporters_cli_tokenizers(self, task: str, model_type: str):
171175
if number_of_tokenizers == 1:
172176
self.assertTrue("Detokenizer is not supported, convert tokenizer only." in output, output)
173177

174-
if task.startswith("text-generation"):
175-
self.assertFalse("Set padding side to left" in output, output)
178+
if task.startswith("text-generation") and compare_versions("openvino-tokenizers", ">=", "2024.3.0.0"):
179+
self.assertTrue("Set tokenizer padding side to left" in output, output)
176180

177181
@parameterized.expand(SUPPORTED_ARCHITECTURES)
178182
def test_exporters_cli_fp16(self, task: str, model_type: str):

0 commit comments

Comments
 (0)