[Bugfix] Honor height/width overrides in LongCatImageEditPipeline (#6222)
Signed-off-by: Yifan Zong <yzong@redhat.com>
This commit is contained in:
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",
|
||||
|
||||
Reference in new issue
Block a user