From e9b0b79132c3653f22a9b41a1c4cc212901717ae Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sat, 8 Mar 2025 11:01:14 +0200
Subject: [PATCH] reset teacache stats

---
 hyvideo/modules/models.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/hyvideo/modules/models.py b/hyvideo/modules/models.py
index 364304f..627152b 100644
--- a/hyvideo/modules/models.py
+++ b/hyvideo/modules/models.py
@@ -1090,6 +1090,7 @@ class HYVideoDiffusionTransformer(ModelMixin, ConfigMixin):
             if self.cnt == 0 or self.cnt == self.num_steps-1:
                 should_calc = True
                 self.accumulated_rel_l1_distance = 0
+                self.teacache_skipped_steps = 0
                 self.previous_modulated_input = modulated_inp.clone()
             else:
                 coefficients = [7.33226126e+02, -4.01131952e+02, 6.75869174e+01, -3.14987800e+00, 9.61237896e-02]