mirror of
https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git
synced 2025-12-16 08:14:32 +08:00
Update model_optimization_nodes.py
This commit is contained in:
parent
c0f9894dc5
commit
6b7eeebe44
@ -796,7 +796,7 @@ def teacache_wanvideo_forward_orig(self, x, t, context, clip_fea=None, freqs=Non
|
||||
else:
|
||||
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
|
||||
x = self.head(x, e)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user