Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-11-04 14:16:57 +00:00
Code Issues Actions 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats 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 = {