[OpenVINO] Set dynamic sequence length for Flux.2 (#1846)

* Fix shapes for transformer

* Apply code-formatting

---------

Co-authored-by: Kazantsev, Roman <roman.kazantsev@intel.com>

Author: openvino-agent

optimum/exporters/openvino/input_generators.py

@@ -745,7 +745,6 @@ def generate(self, input_name: str, framework: str = "pt", int_dtype: str = "int
         if input_name == "img_ids":
             img_ids_height = self.height // 2
             img_ids_width = self.width // 2
-            img_ids_shape = [self.batch_size, img_ids_height * img_ids_width, 3]
             img_ids_shape = (
                 [self.batch_size, img_ids_height * img_ids_width, 3]
                 if is_diffusers_version("<", "0.31.0")

optimum/exporters/openvino/model_configs.py

@@ -2507,11 +2507,13 @@ def inputs(self):
 
         common_inputs["hidden_states"] = {0: "batch_size", 1: "packed_height_width"}
         common_inputs["txt_ids"] = (
-            {0: "batch_size", 1: "sequence_length"} if is_diffusers_version("<", "0.31.0") else {0: "sequence_length"}
+            {0: "batch_size", 1: "sequence_length"}
+            if is_diffusers_version("<", "0.31.0") or self.is_flux2
+            else {0: "sequence_length"}
         )
         common_inputs["img_ids"] = (
             {0: "batch_size", 1: "packed_height_width"}
-            if is_diffusers_version("<", "0.31.0")
+            if is_diffusers_version("<", "0.31.0") or self.is_flux2
             else {0: "packed_height_width"}
         )
         if getattr(self._normalized_config, "guidance_embeds", False):