Skip to content

Commit 2836458

Browse files
Fix test
1 parent 9631027 commit 2836458

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/openvino/test_exporters_cli.py

+2
Original file line numberDiff line numberDiff line change
@@ -375,6 +375,8 @@ def test_exporters_cli_int8(self, task: str, model_type: str):
375375

376376
expected_int8 = _ARCHITECTURES_TO_EXPECTED_INT8[model_type]
377377
expected_int8 = [{"int8": it} for it in expected_int8]
378+
if task.startswith("text2text-generation") and (not task.endswith("with-past") or model.decoder.stateful):
379+
expected_int8 = expected_int8[:2]
378380
check_compression_state_per_model(self, models, expected_int8)
379381

380382
@parameterized.expand(SUPPORTED_SD_HYBRID_ARCHITECTURES)

0 commit comments

Comments
 (0)