diff --git a/hyvideo/modules/models.py b/hyvideo/modules/models.py index 766eacf..aa00331 100644 --- a/hyvideo/modules/models.py +++ b/hyvideo/modules/models.py @@ -758,7 +758,8 @@ class HYVideoDiffusionTransformer(ModelMixin, ConfigMixin): self.teacache_end_step = 100 self.rel_l1_thresh = 0.15 self.accumulated_rel_l1_distance = 0 - self.previous_modulated_input = None + self.previous_modulated_input_cond = None + self.previous_modulated_input_uncond = None self.previous_residual = None self.teacache_device = None diff --git a/nodes.py b/nodes.py index f8bc546..abcfa3e 100644 --- a/nodes.py +++ b/nodes.py @@ -1423,7 +1423,8 @@ class HyVideoSampler: transformer.teacache_skipped_steps_cond = 0 transformer.teacache_skipped_steps_uncond = 0 transformer.accumulated_rel_l1_distance = 0 - transformer.previous_modulated_input = None + transformer.previous_modulated_input_cond = None + transformer.previous_modulated_input_uncond = None transformer.previous_residual = None transformer.teacache_device = device transformer.enable_teacache = True