1 parent 362f10a commit 90ed142Copy full SHA for 90ed142
1 file changed
src/diffusers/models/transformers/transformer_dream.py
@@ -6,10 +6,10 @@
6
import torch.nn as nn
7
import torch.nn.functional as F
8
9
-from ..activations import get_activation
10
from ...configuration_utils import ConfigMixin, register_to_config
11
from ...utils import logging
12
from ...utils.torch_utils import maybe_allow_in_graph
+from ..activations import get_activation
13
from ..attention import AttentionModuleMixin
14
from ..modeling_outputs import Transformer2DModelOutput
15
from ..modeling_utils import ModelMixin
0 commit comments