mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-02-19 06:30:07 +00:00
Fix some custom nodes. (#12455)
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user