Fix condition

huggingface · Dec 12, 2024 · 6aefc75 · 6aefc75
1 parent 54f443d
commit 6aefc75
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/optimum/commands/export/openvino.py b/optimum/commands/export/openvino.py
@@ -415,10 +415,8 @@ def run(self):
             if not self.args.disable_convert_tokenizer:
                 maybe_convert_tokenizers(library_name, self.args.output, model, task=task)
         elif (
-            (task.startswith("text-generation") or task == "automatic-speech-recognition")
-            and quantize_with_dataset
-            or task == "image-text-to-text"
-        ):
+            task.startswith("text-generation") or task in ["automatic-speech-recognition", "image-text-to-text"]
+        ) and quantize_with_dataset:
             if task.startswith("text-generation"):
                 from optimum.intel import OVModelForCausalLM