xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-05-26 23:25:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix TeaCache start/end

Browse Source
This commit is contained in:
kijai 2025-03-17 09:48:06 +02:00
parent 7c488a16ef
commit bb154eb71f
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
nodes/model_optimization_nodes.py
View File

@ -780,7 +780,12 @@ def teacache_wanvideo_forward_orig(self, x, t, context, clip_fea=None, freqs=Non
#print(f"TeaCache: Skipping {suffix} step") #print(f"TeaCache: Skipping {suffix} step")
return should_calc, cache return should_calc, cache
should_calc, cache = tea_cache(x, e0, e, kwargs) teacache_enabled = transformer_options.get("teacache_enabled", False)
if not teacache_enabled:
should_calc = True
else:
should_calc, cache = tea_cache(x, e0, e, kwargs)
if should_calc: if should_calc:
original_x = x.clone().detach() original_x = x.clone().detach()
patches_replace = transformer_options.get("patches_replace", {}) patches_replace = transformer_options.get("patches_replace", {})
@ -796,7 +801,8 @@ def teacache_wanvideo_forward_orig(self, x, t, context, clip_fea=None, freqs=Non
else: else:
x = block(x, e=e0, freqs=freqs, context=context) x = block(x, e=e0, freqs=freqs, context=context)
cache['previous_residual'] = (x - original_x).to(transformer_options["teacache_device"]) if teacache_enabled:
cache['previous_residual'] = (x - original_x).to(transformer_options["teacache_device"])
# head # head
x = self.head(x, e) x = self.head(x, e)