From 9a5d192807124a4657504a25c168a3a7c155623d Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Fri, 9 May 2025 16:43:35 +0300 Subject: [PATCH] more TeaCache fixes.. --- hyvideo/modules/models.py | 4 +++- nodes.py | 14 ++++++-------- 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/hyvideo/modules/models.py b/hyvideo/modules/models.py index aa00331..6010bb9 100644 --- a/hyvideo/modules/models.py +++ b/hyvideo/modules/models.py @@ -760,9 +760,11 @@ class HYVideoDiffusionTransformer(ModelMixin, ConfigMixin): self.accumulated_rel_l1_distance = 0 self.previous_modulated_input_cond = None self.previous_modulated_input_uncond = None - self.previous_residual = None + self.previous_residual_cond = None + self.previous_residual_uncond = None self.teacache_device = None + #slg self.slg_single_blocks = None self.slg_double_blocks = None self.slg_start_percent = 0.0 diff --git a/nodes.py b/nodes.py index abcfa3e..252f856 100644 --- a/nodes.py +++ b/nodes.py @@ -1419,15 +1419,13 @@ class HyVideoSampler: # Initialize TeaCache if enabled if teacache_args is not None: - transformer.cnt = 0 - transformer.teacache_skipped_steps_cond = 0 - transformer.teacache_skipped_steps_uncond = 0 - transformer.accumulated_rel_l1_distance = 0 - transformer.previous_modulated_input_cond = None - transformer.previous_modulated_input_uncond = None - transformer.previous_residual = None - transformer.teacache_device = device transformer.enable_teacache = True + transformer.cnt = 0 + transformer.accumulated_rel_l1_distance = 0 + transformer.teacache_skipped_steps_cond = transformer.teacache_skipped_steps_uncond =0 + transformer.previous_modulated_input_cond = transformer.previous_modulated_input_uncond = None + transformer.previous_residual_cond = transformer.previous_residual_uncond = None + transformer.teacache_device = device transformer.num_steps = steps transformer.rel_l1_thresh = teacache_args["rel_l1_thresh"] transformer.teacache_start_step = teacache_args["start_step"]