revert multiple class mapping

Author: eaidova

optimum/exporters/openvino/model_configs.py

@@ -46,7 +46,7 @@
     WhisperOnnxConfig,
 )
 from optimum.exporters.onnx.model_patcher import ModelPatcher
-from optimum.exporters.tasks import TasksManager
+from optimum.exporters.tasks import TasksManager, get_transformers_tasks_to_model_mapping
 from optimum.utils import DEFAULT_DUMMY_SHAPES
 from optimum.utils.input_generators import (
     DTYPE_MAPPER,
@@ -132,16 +132,14 @@ def init_model_configs():
         "transformers",
         "Qwen2VLForConditionalGeneration",
     )
-
-    TasksManager._TRANSFORMERS_TASKS_TO_MODEL_LOADERS["image-text-to-text"] = (
-        (
-            "AutoModelForImageTextToText",
-            "AutoModelForCausalLM",
-        )
-        if is_transformers_version(">=", "4.46")
-        else TasksManager._TRANSFORMERS_TASKS_TO_MODEL_LOADERS["text-generation"]
+    TasksManager._CUSTOM_CLASSES[("pt", "qwen2-vl", "image-text-to-text")] = (
+        "transformers",
+        "AutoModelForImageTextToText",
     )
 
+    TasksManager._TRANSFORMERS_TASKS_TO_MODEL_LOADERS[
+        "image-text-to-text"
+    ] = TasksManager._TRANSFORMERS_TASKS_TO_MODEL_LOADERS["text-generation"]
     if is_diffusers_available() and "fill" not in TasksManager._DIFFUSERS_TASKS_TO_MODEL_LOADERS:
         TasksManager._DIFFUSERS_TASKS_TO_MODEL_LOADERS["fill"] = "FluxFillPipeline"
         TasksManager._DIFFUSERS_TASKS_TO_MODEL_MAPPINGS["fill"] = {"flux": "FluxFillPipeline"}
@@ -2573,7 +2571,7 @@ def generate(self, input_name: str, framework: str = "pt", int_dtype: str = "int
             return self.random_float_tensor([grid_h * grid_t * grid_w, dim], framework=framework, dtype=float_dtype)
 
         if input_name == "window_index":
-            if self.spatial_merge_size:
+            if self.spatial_merge_size is None:
                 raise ValueError(
                     "`spatial_merge_size` parameter is not found in model config. Can not generate dummy input data for `window_index` input"
                 )