Fix some custom nodes. (#12455)

2026-02-13 17:21:10 -08:00 · 2026-02-13 17:21:10 -08:00 · 726af73867
parent 831351a29e
commit 726af73867
1 changed files with 2 additions and 0 deletions
--- a/comfy/ldm/flux/layers.py
+++ b/comfy/ldm/flux/layers.py
@ -6,6 +6,8 @@ from torch import Tensor, nn

 from .math import attention, rope

+# Fix import for some custom nodes, TODO: delete eventually.
+RMSNorm = None

 class EmbedND(nn.Module):
    def __init__(self, dim: int, theta: int, axes_dim: list):