search

Weifeng-Chen / control-a-video

Official Implementation of "Control-A-Video: Controllable Text-to-Video Generation with Diffusion Models"
GNU General Public License v3.0
359 stars 26 forks source link

Update diffusers package #11

Open pravdomil opened 1 year ago

pravdomil commented 1 year ago

diffusers~=0.17.1

diff --git a/control_a_video/video_diffusion/models/attention.py b/control_a_video/video_diffusion/models/attention.py
index 9fac270..d554184 100755
--- a/control_a_video/video_diffusion/models/attention.py
+++ b/control_a_video/video_diffusion/models/attention.py
@@ -20,11 +20,12 @@

 from diffusers.configuration_utils import ConfigMixin, register_to_config
 from diffusers.models.modeling_utils import ModelMixin
-from diffusers.models.attention import FeedForward, CrossAttention, AdaLayerNorm
+from diffusers.models.attention import FeedForward, AdaLayerNorm
 from diffusers.utils import BaseOutput
 from diffusers.utils.import_utils import is_xformers_available
 from diffusers.models.cross_attention import XFormersCrossAttnProcessor
 from einops import rearrange
+from diffusers.models.cross_attention import CrossAttention

 @dataclass