Reduce img size in dummy_inputs for FLUX (#1070)

* Reduce img size in dummy_inputs for FLUX * change dims in function signature * Update optimum/exporters/openvino/model_configs.py --------- Co-authored-by: Ilyas Moutawwakil <[email protected]>
huggingface · Dec 17, 2024 · 3c229fc · 3c229fc
1 parent f030583
commit 3c229fc
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/optimum/exporters/openvino/model_configs.py b/optimum/exporters/openvino/model_configs.py
@@ -1804,8 +1804,9 @@ def __init__(
         normalized_config: NormalizedVisionConfig,
         batch_size: int = DEFAULT_DUMMY_SHAPES["batch_size"],
         num_channels: int = DEFAULT_DUMMY_SHAPES["num_channels"],
-        width: int = DEFAULT_DUMMY_SHAPES["width"],
-        height: int = DEFAULT_DUMMY_SHAPES["height"],
+        width: int = DEFAULT_DUMMY_SHAPES["width"] // 4,
+        height: int = DEFAULT_DUMMY_SHAPES["height"] // 4,
+        # Reduce img shape by 4 for FLUX to reduce memory usage on conversion
         **kwargs,
     ):
         super().__init__(task, normalized_config, batch_size, num_channels, width, height, **kwargs)