Skip to content

Commit 701f49f

Browse files
author
J石页
committed
NPU adaption for RMSNorm
1 parent 2fb3f9f commit 701f49f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/normalization.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
import torch.nn as nn
2121
import torch.nn.functional as F
2222

23-
from ..utils import is_torch_version, is_torch_npu_available
23+
from ..utils import is_torch_npu_available, is_torch_version
2424
from .activations import get_activation
2525
from .embeddings import CombinedTimestepLabelEmbeddings, PixArtAlphaCombinedTimestepSizeEmbeddings
2626

0 commit comments

Comments
 (0)