diff --git a/animatediff/motion_module_ad.py b/animatediff/motion_module_ad.py
index 1830266..b66538e 100644
--- a/animatediff/motion_module_ad.py
+++ b/animatediff/motion_module_ad.py
@@ -920,6 +920,7 @@ def set_cameractrl_effect(self, multival: Union[float, Tensor]):
         self.temp_cameractrl_effect = None
 
     def set_sub_idxs(self, sub_idxs: list[int]):
+        self.sub_idxs = sub_idxs
         for block in self.transformer_blocks:
             block.set_sub_idxs(sub_idxs)
 
diff --git a/pyproject.toml b/pyproject.toml
index 35964e5..ce99f35 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-animatediff-evolved"
 description = "Improved AnimateDiff integration for ComfyUI."
-version = "1.1.3"
+version = "1.1.4"
 license = { file = "LICENSE" }
 dependencies = []