diff --git a/comfy_extras/nodes_flux.py b/comfy_extras/nodes_flux.py index 539635bb2..fb4811f84 100644 --- a/comfy_extras/nodes_flux.py +++ b/comfy_extras/nodes_flux.py @@ -125,7 +125,7 @@ class ReduxApplyWithAttnMask: m.set_model_patch_replace(wrapper, "dit", "double_blocks", i) else: m = model - return (c, m) + return (m, c) NODE_CLASS_MAPPINGS = { "CLIPTextEncodeFlux": CLIPTextEncodeFlux,