|
50 | 50 | from optimum.intel.utils.import_utils import ( |
51 | 51 | compare_versions, |
52 | 52 | is_openvino_tokenizers_available, |
| 53 | + is_openvino_version, |
53 | 54 | is_tokenizers_version, |
54 | 55 | is_transformers_version, |
55 | 56 | ) |
@@ -80,20 +81,20 @@ class OVCLIExportTestCase(unittest.TestCase): |
80 | 81 | if is_transformers_version(">=", "4.45"): |
81 | 82 | SUPPORTED_ARCHITECTURES.extend([("text-to-image", "stable-diffusion-3"), ("text-to-image", "flux")]) |
82 | 83 | EXPECTED_NUMBER_OF_TOKENIZER_MODELS = { |
83 | | - "gpt2": 2 if is_tokenizers_version("<", "0.20") else 0, |
| 84 | + "gpt2": 2 if is_tokenizers_version("<", "0.20") or is_openvino_version(">=", "2024.5") else 0, |
84 | 85 | "t5": 0, # no .model file in the repository |
85 | 86 | "albert": 0, # not supported yet |
86 | 87 | "distilbert": 1, # no detokenizer |
87 | | - "roberta": 2 if is_tokenizers_version("<", "0.20") else 0, |
| 88 | + "roberta": 2 if is_tokenizers_version("<", "0.20") or is_openvino_version(">=", "2024.5") else 0, |
88 | 89 | "vit": 0, # no tokenizer for image model |
89 | 90 | "wav2vec2": 0, # no tokenizer |
90 | 91 | "bert": 1, # no detokenizer |
91 | | - "blenderbot": 2 if is_tokenizers_version("<", "0.20") else 0, |
92 | | - "stable-diffusion": 2 if is_tokenizers_version("<", "0.20") else 0, |
93 | | - "stable-diffusion-xl": 4 if is_tokenizers_version("<", "0.20") else 0, |
94 | | - "stable-diffusion-3": 6 if is_tokenizers_version("<", "0.20") else 2, |
95 | | - "flux": 4 if is_tokenizers_version("<", "0.20") else 0, |
96 | | - "llava": 2 if is_tokenizers_version("<", "0.20") else 0, |
| 92 | + "blenderbot": 2 if is_tokenizers_version("<", "0.20") or is_openvino_version(">=", "2024.5") else 0, |
| 93 | + "stable-diffusion": 2 if is_tokenizers_version("<", "0.20") or is_openvino_version(">=", "2024.5") else 0, |
| 94 | + "stable-diffusion-xl": 4 if is_tokenizers_version("<", "0.20") or is_openvino_version(">=", "2024.5") else 0, |
| 95 | + "stable-diffusion-3": 6 if is_tokenizers_version("<", "0.20") or is_openvino_version(">=", "2024.5") else 2, |
| 96 | + "flux": 4 if is_tokenizers_version("<", "0.20") or is_openvino_version(">=", "2024.5") else 0, |
| 97 | + "llava": 2 if is_tokenizers_version("<", "0.20") or is_openvino_version(">=", "2024.5") else 0, |
97 | 98 | } |
98 | 99 |
|
99 | 100 | SUPPORTED_SD_HYBRID_ARCHITECTURES = [ |
|
0 commit comments