Skip to content

Commit f7753b1

Browse files
authored
more update in modular (#12560)
* move node registry to mellon * up * fix * modula rpipeline update: filter out none for input_names, fix default blocks for pipe.init() and allow user pass additional kwargs_type in a dict * qwen modular refactor, unpack before decode * update mellon node config, adding* to required_inputs and required_model_inputs * modularpipeline.from_pretrained: error out if no config found * add a component_names property to modular blocks to be consistent! * flux image_encoder -> vae_encoder * controlnet_bundle * refator MellonNodeConfig MellonPipelineConfig * refactor & simplify mellon utils * vae_image_encoder -> vae_encoder * mellon config save keep key order * style + copies * add kwargs input for zimage
1 parent b530968 commit f7753b1

File tree

9 files changed

+575
-864
lines changed

9 files changed

+575
-864
lines changed

src/diffusers/modular_pipelines/flux/modular_blocks.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -360,7 +360,7 @@ def description(self):
360360
AUTO_BLOCKS = InsertableDict(
361361
[
362362
("text_encoder", FluxTextEncoderStep()),
363-
("image_encoder", FluxAutoVaeEncoderStep()),
363+
("vae_encoder", FluxAutoVaeEncoderStep()),
364364
("denoise", FluxCoreDenoiseStep()),
365365
("decode", FluxDecodeStep()),
366366
]
@@ -369,7 +369,7 @@ def description(self):
369369
AUTO_BLOCKS_KONTEXT = InsertableDict(
370370
[
371371
("text_encoder", FluxTextEncoderStep()),
372-
("image_encoder", FluxKontextAutoVaeEncoderStep()),
372+
("vae_encoder", FluxKontextAutoVaeEncoderStep()),
373373
("denoise", FluxKontextCoreDenoiseStep()),
374374
("decode", FluxDecodeStep()),
375375
]

0 commit comments

Comments
 (0)