This commit is contained in:
comfyanonymous 2025-11-14 01:20:34 -05:00
parent 728db5c47c
commit c6414ed182

View File

@ -1,12 +1,9 @@
import torch import torch
from torch import Tensor, nn from torch import Tensor, nn
from comfy.ldm.flux.math import attention
from comfy.ldm.flux.layers import ( from comfy.ldm.flux.layers import (
MLPEmbedder, MLPEmbedder,
RMSNorm, RMSNorm,
QKNorm,
SelfAttention,
ModulationOut, ModulationOut,
) )