Skip to content

Commit fb4a007

Browse files
authored
recover scheduler infer_step (#891)
1 parent a798753 commit fb4a007

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

lightx2v/models/schedulers/scheduler.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,8 @@ def __init__(self, config):
66
self.config = config
77
self.latents = None
88
self.step_index = 0
9+
self.infer_steps = config["infer_steps"]
10+
self.caching_records = [True] * config["infer_steps"]
911
self.flag_df = False
1012
self.transformer_infer = None
1113
self.infer_condition = True # cfg status

0 commit comments

Comments
 (0)