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

Fix TeaCache end step

Browse Source
This commit is contained in:
kijai
2025-04-17 01:18:42 +03:00
parent 761d188dba
commit 8257cd1f8a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
wanvideo/modules/model.py
+1 -1
View File
@@ -1157,7 +1157,7 @@ class WanModel(ModelMixin, ConfigMixin):
if b <= self.blocks_to_swap and self.blocks_to_swap >= 0: if b <= self.blocks_to_swap and self.blocks_to_swap >= 0:
block.to(self.offload_device, non_blocking=self.use_non_blocking) block.to(self.offload_device, non_blocking=self.use_non_blocking)
if self.enable_teacache and pred_id is not None: if self.enable_teacache and (self.teacache_start_step <= current_step <= self.teacache_end_step) and pred_id is not None:
self.teacache_state.update( self.teacache_state.update(
pred_id, pred_id,
previous_residual=(x.to(original_x.device) - original_x), previous_residual=(x.to(original_x.device) - original_x),