Skip to content

Commit a5c108b

Browse files
nikita-savelyevvmvafin
authored andcommitted
Update OV test references (#1417)
1 parent b07924b commit a5c108b

File tree

3 files changed

+9
-3
lines changed

3 files changed

+9
-3
lines changed

tests/openvino/test_exporters_cli.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -442,7 +442,10 @@ class OVCLIExportTestCase(unittest.TestCase):
442442
"--dataset c4 --num-samples 1",
443443
{"encoder": 30, "decoder": 52, "decoder_with_past": 61}
444444
if is_transformers_version("<=", "4.36.0")
445-
else {"encoder": 30, "decoder": 62 if is_openvino_version("<", "2025.3") else 52},
445+
else {
446+
"encoder": 30,
447+
"decoder": 62 if is_nncf_version("<=", "2.17") and is_openvino_version("<", "2025.3") else 52,
448+
},
446449
(
447450
{"encoder": {"int8": 32}, "decoder": {"int8": 52}, "decoder_with_past": {"int8": 42}}
448451
if is_transformers_version("<=", "4.36.0")

tests/openvino/test_quantization.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -394,7 +394,10 @@ class OVQuantizerTest(unittest.TestCase):
394394
),
395395
{"encoder": 30, "decoder": 52, "decoder_with_past": 61}
396396
if is_transformers_version("<=", "4.36.0")
397-
else {"encoder": 30, "decoder": 62},
397+
else {
398+
"encoder": 30,
399+
"decoder": 62 if is_nncf_version("<=", "2.17") and is_openvino_version("<", "2025.3") else 52,
400+
},
398401
(
399402
{"encoder": {"int8": 32}, "decoder": {"int8": 52}, "decoder_with_past": {"int8": 42}}
400403
if is_transformers_version("<=", "4.36.0")

tests/openvino/utils_tests.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -304,7 +304,7 @@
304304
"transformer": 58,
305305
"vae_decoder": 28,
306306
"vae_encoder": 28,
307-
"text_encoder": 18 if is_nncf_version("<", "2.18.0") else 16,
307+
"text_encoder": 16 if is_nncf_version(">", "2.17") else 18,
308308
},
309309
"ltx-video": {
310310
"transformer": 34,

0 commit comments

Comments
 (0)