Skip to content

Commit 3c229fc

Browse files
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 <57442720+IlyasMoutawwakil@users.noreply.github.com>
1 parent f030583 commit 3c229fc

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

optimum/exporters/openvino/model_configs.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -1804,8 +1804,9 @@ def __init__(
18041804
normalized_config: NormalizedVisionConfig,
18051805
batch_size: int = DEFAULT_DUMMY_SHAPES["batch_size"],
18061806
num_channels: int = DEFAULT_DUMMY_SHAPES["num_channels"],
1807-
width: int = DEFAULT_DUMMY_SHAPES["width"],
1808-
height: int = DEFAULT_DUMMY_SHAPES["height"],
1807+
width: int = DEFAULT_DUMMY_SHAPES["width"] // 4,
1808+
height: int = DEFAULT_DUMMY_SHAPES["height"] // 4,
1809+
# Reduce img shape by 4 for FLUX to reduce memory usage on conversion
18091810
**kwargs,
18101811
):
18111812
super().__init__(task, normalized_config, batch_size, num_channels, width, height, **kwargs)

0 commit comments

Comments
 (0)