comi-logic/Wan2GP
7
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix tea cache disabled with diffusion forcing

Browse Source
This commit is contained in:
deepbeepmeep
2025-07-04 13:19:02 +02:00
committed by GitHub
parent eb811e0c52
commit d8a829bab2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
wan/diffusion_forcing.py
View File

@@ -334,7 +334,7 @@ class DTT2V:
self.model.compute_teacache_threshold(self.model.cache_start_step, time_steps_comb, self.model.cache_multiplier) self.model.compute_teacache_threshold(self.model.cache_start_step, time_steps_comb, self.model.cache_multiplier)
del time_steps_comb del time_steps_comb
else: else:
trans.enable_cache == None self.model.enable_cache = None
from mmgp import offload from mmgp import offload
freqs = get_rotary_pos_embed(latents.shape[1 :], enable_RIFLEx= False) freqs = get_rotary_pos_embed(latents.shape[1 :], enable_RIFLEx= False)
kwrags = { kwrags = {