[Bugfix] Honor height/width overrides in LongCatImageEditPipeline (#6222)

Signed-off-by: Yifan Zong <yzong@redhat.com>
This commit is contained in:
yzong-rh authored and GitHub committed 2026-08-25 02:09:00 +00:00
1 parent 53b60bd9bc
commit 710024fa6f
4 files changed
+37 -14

No files matched your search

@@ -200,6 +200,22 @@ def tiny_longcat_image_builder() -> str:
)
def tiny_longcat_image_edit_builder() -> str:
return build_tiny_from_configs(
"LongCatImageEditPipeline",
"meituan-longcat/LongCat-Image-Edit",
transform={
"text_encoder": _shrink_qwen_text_encoder_config,
"transformer": partial(
_shrink_dit_rope_config,
num_single_layers=2,
default_axes_dims_rope=[16, 56, 56],
joint_attention_dim=64,
),
},
)
def tiny_flux_builder() -> str:
return build_tiny_from_configs(
"FluxPipeline",