Open pravdomil opened 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
diffusers~=0.17.1