Skip to content

Commit c441dac

Browse files
Switch test to fp8
1 parent 1d9f2f8 commit c441dac

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/openvino/test_exporters_cli.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -196,12 +196,12 @@ class OVCLIExportTestCase(unittest.TestCase):
196196
(
197197
"automatic-speech-recognition-with-past",
198198
"whisper",
199-
"int8",
199+
"f8e4m3",
200200
"--dataset librispeech --num-samples 1 --smooth-quant-alpha 0.9 --trust-remote-code",
201201
[14, 22, 21] if is_transformers_version("<=", "4.36.0") else [14, 22, 25],
202-
[{"int8": 14}, {"int8": 21}, {"int8": 17}]
202+
[{"f8e4m3": 14}, {"f8e4m3": 21}, {"f8e4m3": 17}]
203203
if is_transformers_version("<=", "4.36.0")
204-
else [{"int8": 14}, {"int8": 22}, {"int8": 18}],
204+
else [{"f8e4m3": 14}, {"f8e4m3": 22}, {"f8e4m3": 18}],
205205
),
206206
(
207207
"text-generation",

0 commit comments

Comments
 (0)