Add a way for nodes to add pre attn patches to flux model. (#12861)

2026-03-09 20:37:58 -07:00
parent 740d998c9c
commit c4fb0271cd
3 changed files with 17 additions and 2 deletions
@@ -170,7 +170,7 @@ class Flux(nn.Module):

        if "post_input" in patches:
            for p in patches["post_input"]:
-                out = p({"img": img, "txt": txt, "img_ids": img_ids, "txt_ids": txt_ids})
+                out = p({"img": img, "txt": txt, "img_ids": img_ids, "txt_ids": txt_ids, "transformer_options": transformer_options})
                img = out["img"]
                txt = out["txt"]
                img_ids = out["img_ids"]