From b37ed57f20e93b721bcf954e552e449466c81d8d Mon Sep 17 00:00:00 2001
From: wangshankun <wangshankun2011@hotmail.com>
Date: Tue, 10 Feb 2026 10:54:52 +0000
Subject: [PATCH] recover scheduler infer_step

---
 lightx2v/models/schedulers/scheduler.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/lightx2v/models/schedulers/scheduler.py b/lightx2v/models/schedulers/scheduler.py
index b29786f47..a14403d0d 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