|
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