We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a798753 commit fb4a007Copy full SHA for fb4a007
1 file changed
lightx2v/models/schedulers/scheduler.py
@@ -6,6 +6,8 @@ def __init__(self, config):
6
self.config = config
7
self.latents = None
8
self.step_index = 0
9
+ self.infer_steps = config["infer_steps"]
10
+ self.caching_records = [True] * config["infer_steps"]
11
self.flag_df = False
12
self.transformer_infer = None
13
self.infer_condition = True # cfg status
0 commit comments