nodes/kijai-ComfyUI-HunyuanVideoWrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

more TeaCache fixes..

Browse Source
This commit is contained in:
kijai
2025-05-09 16:43:35 +03:00
parent 14cc8ae0a5
commit 9a5d192807
2 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hyvideo/modules/models.py
+3 -1
View File
@@ -760,9 +760,11 @@ class HYVideoDiffusionTransformer(ModelMixin, ConfigMixin):
self.accumulated_rel_l1_distance = 0 self.accumulated_rel_l1_distance = 0
self.previous_modulated_input_cond = None self.previous_modulated_input_cond = None
self.previous_modulated_input_uncond = 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 self.teacache_device = None
#slg
self.slg_single_blocks = None self.slg_single_blocks = None
self.slg_double_blocks = None self.slg_double_blocks = None
self.slg_start_percent = 0.0 self.slg_start_percent = 0.0
nodes.py
+6 -8
View File
@@ -1419,15 +1419,13 @@ class HyVideoSampler:
# Initialize TeaCache if enabled # Initialize TeaCache if enabled
if teacache_args is not None: 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.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.num_steps = steps
transformer.rel_l1_thresh = teacache_args["rel_l1_thresh"] transformer.rel_l1_thresh = teacache_args["rel_l1_thresh"]
transformer.teacache_start_step = teacache_args["start_step"] transformer.teacache_start_step = teacache_args["start_step"]