Fix some custom nodes. (#12455)

This commit is contained in:
comfyanonymous
2026-02-13 17:21:10 -08:00
committed by GitHub
parent 831351a29e
commit 726af73867

View File

@@ -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):