48
48
from optimum .intel .utils .import_utils import (
49
49
compare_versions ,
50
50
is_openvino_tokenizers_available ,
51
+ is_transformers_version ,
51
52
)
52
53
53
54
@@ -73,17 +74,17 @@ class OVCLIExportTestCase(unittest.TestCase):
73
74
("image-to-image" , "stable-diffusion-xl-refiner" ),
74
75
)
75
76
EXPECTED_NUMBER_OF_TOKENIZER_MODELS = {
76
- "gpt2" : 2 ,
77
+ "gpt2" : 2 if is_transformers_version ( "<" , "4.45" ) else 0 ,
77
78
"t5" : 0 , # no .model file in the repository
78
79
"albert" : 0 , # not supported yet
79
80
"distilbert" : 1 , # no detokenizer
80
- "roberta" : 2 ,
81
+ "roberta" : 2 if is_transformers_version ( "<" , "4.45" ) else 0 ,
81
82
"vit" : 0 , # no tokenizer for image model
82
83
"wav2vec2" : 0 , # no tokenizer
83
84
"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 ,
87
88
}
88
89
89
90
SUPPORTED_SD_HYBRID_ARCHITECTURES = (
0 commit comments