diff --git a/tests/openvino/test_exporters_cli.py b/tests/openvino/test_exporters_cli.py index 9320a2f9b..a25a5cb9c 100644 --- a/tests/openvino/test_exporters_cli.py +++ b/tests/openvino/test_exporters_cli.py @@ -95,8 +95,8 @@ class OVCLIExportTestCase(unittest.TestCase): ) EXPECTED_NUMBER_OF_TOKENIZER_MODELS = { "gpt2": 2 if is_tokenizers_version("<", "0.20") or is_openvino_version(">=", "2024.5") else 0, - "t5": 0, # no .model file in the repository - "albert": 0, # not supported yet + "t5": 0 if is_openvino_version("<", "2025.1") else 2, # 2025.1 brings support for unigram tokenizers + "albert": 0 if is_openvino_version("<", "2025.1") else 2, # 2025.1 brings support for unigram tokenizers "distilbert": 1 if is_openvino_version("<", "2025.0") else 2, # no detokenizer before 2025.0 "roberta": 2 if is_tokenizers_version("<", "0.20") or is_openvino_version(">=", "2024.5") else 0, "vit": 0, # no tokenizer for image model diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py index cff248cf5..1d94b72db 100644 --- a/tests/openvino/test_modeling.py +++ b/tests/openvino/test_modeling.py @@ -149,7 +149,7 @@ def __init__(self, *args, **kwargs): self.OV_MODEL_ID = "echarlaix/distilbert-base-uncased-finetuned-sst-2-english-openvino" self.OV_DECODER_MODEL_ID = "helenai/gpt2-ov" self.OV_SEQ2SEQ_MODEL_ID = "echarlaix/t5-small-openvino" - self.OV_SD_DIFFUSION_MODEL_ID = "hf-internal-testing/tiny-stable-diffusion-openvino" + self.OV_SD_DIFFUSION_MODEL_ID = "katuni4ka/tiny-stable-diffusion-openvino" self.OV_FLUX_DIFFUSION_MODEL_ID = "katuni4ka/tiny-random-flux-ov" self.OV_VLM_MODEL_ID = "katuni4ka/tiny-random-llava-ov"