TeaCache fixes

2025-05-09 16:36:00 +03:00
parent 0d9fe9b86a
commit 14cc8ae0a5
2 changed files with 4 additions and 2 deletions
@@ -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

@@ -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