Implement NAG on all the models based on the Flux code. (#12500)

Use the Normalized Attention Guidance node. Flux, Flux2, Klein, Chroma, Chroma radiance, Hunyuan Video, etc..
2026-04-28 18:31:31 +00:00 · 2026-02-16 20:30:34 -08:00
parent 8a6fbc2dc2
commit 18927538a1
7 changed files with 128 additions and 1 deletions
--- a/nodes.py
+++ b/nodes.py
@@ -2437,6 +2437,7 @@ async def init_builtin_extra_nodes():
        "nodes_color.py",
        "nodes_toolkit.py",
        "nodes_replacements.py",
+        "nodes_nag.py",
    ]

    import_failed = []