From bc093d13d6108163ae9da64eb227962ff056ba86 Mon Sep 17 00:00:00 2001 From: eaidova Date: Tue, 18 Mar 2025 09:34:46 +0400 Subject: [PATCH] fix tests running with nightly --- tests/openvino/test_exporters_cli.py | 4 ++-- tests/openvino/test_modeling.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/openvino/test_exporters_cli.py b/tests/openvino/test_exporters_cli.py index 9320a2f9bf..a25a5cb9c8 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 cff248cf5a..1d94b72db1 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"