From 7eeed536d13dff58c7a8093b7d2f10852d77b3b8 Mon Sep 17 00:00:00 2001
From: Jedrzej Kosinski <kosinkadink1@gmail.com>
Date: Fri, 16 Aug 2024 09:12:12 -0500
Subject: [PATCH] Fixed scale_masks param being None with CameraCtrl

---
 animatediff/motion_module_ad.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/animatediff/motion_module_ad.py b/animatediff/motion_module_ad.py
index 19fda16..1830266 100644
--- 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: