Fix automatic-speech-recognition-with-past quantization from CLI

Author: nikita-savelyevv

optimum/commands/export/openvino.py

@@ -429,7 +429,7 @@ def run(self):
                 maybe_convert_tokenizers(library_name, self.args.output, model, task=task)
         elif (
             quantize_with_dataset
-            and (task.startswith("text-generation") or task == "automatic-speech-recognition")
+            and (task.startswith("text-generation") or "automatic-speech-recognition" in task)
             or (task == "image-text-to-text" and quantization_config is not None)
         ):
             if task.startswith("text-generation"):

optimum/intel/openvino/utils.py

@@ -134,6 +134,7 @@
     "open_clip_vision": "OVModelOpenCLIPVisual",
     "open_clip": "OVModelOpenCLIPForZeroShotImageClassification",
     "automatic-speech-recognition": "OVModelForSpeechSeq2Seq",
+    "automatic-speech-recognition-with-past": "OVModelForSpeechSeq2Seq",
 }
 
 

tests/openvino/test_exporters_cli.py

@@ -193,6 +193,16 @@ class OVCLIExportTestCase(unittest.TestCase):
             if is_transformers_version("<=", "4.36.0")
             else [{"int8": 14}, {"int8": 22}, {"int8": 18}],
         ),
+        (
+            "automatic-speech-recognition-with-past",
+            "whisper",
+            "int8",
+            "--dataset librispeech --num-samples 1 --smooth-quant-alpha 0.9 --trust-remote-code",
+            [14, 22, 21] if is_transformers_version("<=", "4.36.0") else [14, 22, 25],
+            [{"int8": 14}, {"int8": 21}, {"int8": 17}]
+            if is_transformers_version("<=", "4.36.0")
+            else [{"int8": 14}, {"int8": 22}, {"int8": 18}],
+        ),
         (
             "text-generation",
             "llama",
@@ -694,13 +704,14 @@ def test_exporters_cli_full_quantization(
     ):
         with TemporaryDirectory() as tmpdir:
             subprocess.run(
-                f"optimum-cli export openvino --model {MODEL_NAMES[model_type]} --quant-mode {quant_mode} {option} {tmpdir}",
+                f"optimum-cli export openvino --task {task} --model {MODEL_NAMES[model_type]} "
+                f"--quant-mode {quant_mode} {option} {tmpdir}",
                 shell=True,
                 check=True,
             )
             model = eval(_HEAD_TO_AUTOMODELS[task]).from_pretrained(tmpdir)
 
-            if task == "automatic-speech-recognition":
+            if "automatic-speech-recognition" in task:
                 submodels = [model.encoder, model.decoder]
                 if model.decoder_with_past is not None:
                     submodels.append(model.decoder_with_past)