xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-23 03:34:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update model_optimization_nodes.py

Browse Source
This commit is contained in:
kijai 2025-03-16 13:15:47 +02:00
parent c0f9894dc5
commit 6b7eeebe44
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nodes/model_optimization_nodes.py
View File

@ -796,7 +796,7 @@ 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"]) cache['previous_residual'] = (x - original_x).to(transformer_options["teacache_device"])
# head # head
x = self.head(x, e) x = self.head(x, e)