Add keyframe example

hyvideo/modules/models.py

@@ -1090,7 +1090,6 @@ 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]

nodes.py

@@ -1338,6 +1338,7 @@ class HyVideoSampler:
                     transformer.last_frame_count != num_frames):
                 # Reset TeaCache state on dimension change
                 transformer.cnt = 0
+                transformer.teacache_skipped_steps = 0
                 transformer.accumulated_rel_l1_distance = 0
                 transformer.previous_modulated_input = None
                 transformer.previous_residual = None
@@ -1405,6 +1406,7 @@ class HyVideoSampler:
 
         if teacache_args is not None:
             log.info(f"TeaCache skipped {transformer.teacache_skipped_steps} steps")
+            transformer.teacache_skipped_steps = 0
 
         if force_offload:
             if model["manual_offloading"]: