Spaces:
Running
on
Zero
Running
on
Zero
Update latentsync/models/attention.py
Browse files
latentsync/models/attention.py
CHANGED
@@ -12,7 +12,7 @@ from diffusers.configuration_utils import ConfigMixin, register_to_config
|
|
12 |
from diffusers.models.modeling_utils import ModelMixin
|
13 |
from diffusers.utils import BaseOutput
|
14 |
from diffusers.utils.import_utils import is_xformers_available
|
15 |
-
from diffusers.models.attention import CrossAttention, FeedForward, AdaLayerNorm
|
16 |
|
17 |
from einops import rearrange, repeat
|
18 |
from .utils import zero_module
|
|
|
12 |
from diffusers.models.modeling_utils import ModelMixin
|
13 |
from diffusers.utils import BaseOutput
|
14 |
from diffusers.utils.import_utils import is_xformers_available
|
15 |
+
from diffusers.models.attention import Attention as CrossAttention, FeedForward, AdaLayerNorm
|
16 |
|
17 |
from einops import rearrange, repeat
|
18 |
from .utils import zero_module
|