Skip to content

Commit 929f846

Browse files
committed
fix tests running with nightly
1 parent 5ac3544 commit 929f846

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

tests/openvino/test_exporters_cli.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -95,8 +95,8 @@ class OVCLIExportTestCase(unittest.TestCase):
9595
)
9696
EXPECTED_NUMBER_OF_TOKENIZER_MODELS = {
9797
"gpt2": 2 if is_tokenizers_version("<", "0.20") or is_openvino_version(">=", "2024.5") else 0,
98-
"t5": 0, # no .model file in the repository
99-
"albert": 0, # not supported yet
98+
"t5": 0 if is_openvino_version("<", "2025.1") else 2, # 2025.1 brings support for unigram tokenizers
99+
"albert": 0 if is_openvino_version("<", "2025.1") else 2, # 2025.1 brings support for unigram tokenizers
100100
"distilbert": 1 if is_openvino_version("<", "2025.0") else 2, # no detokenizer before 2025.0
101101
"roberta": 2 if is_tokenizers_version("<", "0.20") or is_openvino_version(">=", "2024.5") else 0,
102102
"vit": 0, # no tokenizer for image model

tests/openvino/test_modeling.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ def __init__(self, *args, **kwargs):
149149
self.OV_MODEL_ID = "echarlaix/distilbert-base-uncased-finetuned-sst-2-english-openvino"
150150
self.OV_DECODER_MODEL_ID = "helenai/gpt2-ov"
151151
self.OV_SEQ2SEQ_MODEL_ID = "echarlaix/t5-small-openvino"
152-
self.OV_SD_DIFFUSION_MODEL_ID = "hf-internal-testing/tiny-stable-diffusion-openvino"
152+
self.OV_SD_DIFFUSION_MODEL_ID = "katuni4ka/tiny-stable-diffusion-openvino"
153153
self.OV_FLUX_DIFFUSION_MODEL_ID = "katuni4ka/tiny-random-flux-ov"
154154
self.OV_VLM_MODEL_ID = "katuni4ka/tiny-random-llava-ov"
155155

0 commit comments

Comments
 (0)