Fixed scale_masks param being None with CameraCtrl

Kosinkadink · Aug 16, 2024 · 7eeed53 · 7eeed53
1 parent 541ece5
commit 7eeed53
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/animatediff/motion_module_ad.py b/animatediff/motion_module_ad.py
@@ -1148,6 +1148,8 @@ def forward(
         view_options: Union[ContextOptions, None]=None,
         mm_kwargs: dict[str]=None,
     ):
+        if scale_masks is None:
+            scale_masks = [None] * len(self.attention_blocks)
         # make view_options None if context_length > video_length, or if equal and equal not allowed
         if view_options:
             if view_options.context_length > video_length: