optimize VRAM

2025-03-12 19:39:57 +08:00
parent be73edb3ed
commit 2e75a8b611
2 changed files with 1 additions and 1 deletions
@@ -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)