From d8a829bab2f0a548d655a3d27d671369bba12480 Mon Sep 17 00:00:00 2001
From: deepbeepmeep <84379123+deepbeepmeep@users.noreply.github.com>
Date: Fri, 4 Jul 2025 13:19:02 +0200
Subject: [PATCH] Fix tea cache disabled with diffusion forcing

---
 wan/diffusion_forcing.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/wan/diffusion_forcing.py b/wan/diffusion_forcing.py
index 4a737f1..c9b735c 100644
--- a/wan/diffusion_forcing.py
+++ b/wan/diffusion_forcing.py
@@ -334,7 +334,7 @@ class DTT2V:
             self.model.compute_teacache_threshold(self.model.cache_start_step, time_steps_comb, self.model.cache_multiplier)
             del time_steps_comb
         else:
-            trans.enable_cache == None
+            self.model.enable_cache = None
         from mmgp import offload
         freqs = get_rotary_pos_embed(latents.shape[1 :], enable_RIFLEx= False) 
         kwrags = {