diff --git a/lightx2v/models/schedulers/scheduler.py b/lightx2v/models/schedulers/scheduler.py
index b29786f4..a14403d0 100755
--- a/lightx2v/models/schedulers/scheduler.py
+++ b/lightx2v/models/schedulers/scheduler.py
@@ -6,6 +6,8 @@ def __init__(self, config):
         self.config = config
         self.latents = None
         self.step_index = 0
+        self.infer_steps = config["infer_steps"]
+        self.caching_records = [True] * config["infer_steps"]
         self.flag_df = False
         self.transformer_infer = None
         self.infer_condition = True  # cfg status