diff --git a/assets/compile.png b/assets/compile.png
index 87594b1..df07009 100644
Binary files a/assets/compile.png and b/assets/compile.png differ
diff --git a/nodes.py b/nodes.py
index 3aa431e..5c3285a 100644
--- a/nodes.py
+++ b/nodes.py
@@ -547,7 +547,7 @@ def teacache_wanmodel_forward_orig(
             ori_x = x.clone()
             for block in self.blocks:
                 x = block(x, **kwargs)
-            self.previous_residual = (x - ori_x).to(torch.device('cpu'))
+            self.previous_residual = (x - ori_x).to(torch.device('cpu')) 
 
         # head
         x = self.head(x, e)