Skip to content

Commit e18f4a3

Browse files
update number of expected exported tokenizers
1 parent b31524c commit e18f4a3

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

tests/openvino/test_exporters_cli.py

+6-5
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,7 @@
4848
from optimum.intel.utils.import_utils import (
4949
compare_versions,
5050
is_openvino_tokenizers_available,
51+
is_transformers_version,
5152
)
5253

5354

@@ -73,17 +74,17 @@ class OVCLIExportTestCase(unittest.TestCase):
7374
("image-to-image", "stable-diffusion-xl-refiner"),
7475
)
7576
EXPECTED_NUMBER_OF_TOKENIZER_MODELS = {
76-
"gpt2": 2,
77+
"gpt2": 2 if is_transformers_version("<", "4.45") else 0,
7778
"t5": 0, # no .model file in the repository
7879
"albert": 0, # not supported yet
7980
"distilbert": 1, # no detokenizer
80-
"roberta": 2,
81+
"roberta": 2 if is_transformers_version("<", "4.45") else 0,
8182
"vit": 0, # no tokenizer for image model
8283
"wav2vec2": 0, # no tokenizer
8384
"bert": 1, # no detokenizer
84-
"blenderbot": 2,
85-
"stable-diffusion": 2,
86-
"stable-diffusion-xl": 4,
85+
"blenderbot": 2 if is_transformers_version("<", "4.45") else 0,
86+
"stable-diffusion": 2 if is_transformers_version("<", "4.45") else 0,
87+
"stable-diffusion-xl": 4 if is_transformers_version("<", "4.45") else 0,
8788
}
8889

8990
SUPPORTED_SD_HYBRID_ARCHITECTURES = (

0 commit comments

Comments
 (0)