wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-26 18:32:35 +08:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

Fix some custom nodes. (#12455)

Browse Source
This commit is contained in:
comfyanonymous 2026-02-13 17:21:10 -08:00 committed by GitHub
parent 831351a29e
commit 726af73867
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/ldm/flux/layers.py
View File

@ -6,6 +6,8 @@ from torch import Tensor, nn
from .math import attention, rope from .math import attention, rope
# Fix import for some custom nodes, TODO: delete eventually.
RMSNorm = None
class EmbedND(nn.Module): class EmbedND(nn.Module):
def __init__(self, dim: int, theta: int, axes_dim: list): def __init__(self, dim: int, theta: int, axes_dim: list):