From dbc726f80c9ac0512d2611fad63d984b8c03886f Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Tue, 29 Apr 2025 17:42:00 -0700
Subject: [PATCH] Better vace memory estimation. (#7875)

---
 comfy/ldm/wan/model.py    | 1 +
 comfy/supported_models.py | 4 ++++
 2 files changed, 5 insertions(+)

diff --git a/comfy/ldm/wan/model.py b/comfy/ldm/wan/model.py
index b8eec3af..66bee748 100644
--- a/comfy/ldm/wan/model.py
+++ b/comfy/ldm/wan/model.py
@@ -631,6 +631,7 @@ class VaceWanModel(WanModel):
             if ii is not None:
                 c_skip, c = self.vace_blocks[ii](c, x=x_orig, e=e0, freqs=freqs, context=context, context_img_len=context_img_len)
                 x += c_skip * vace_strength
+                del c_skip
         # head
         x = self.head(x, e)
 
diff --git a/comfy/supported_models.py b/comfy/supported_models.py
index 5e55035c..69bcee1f 100644
--- a/comfy/supported_models.py
+++ b/comfy/supported_models.py
@@ -993,6 +993,10 @@ class WAN21_Vace(WAN21_T2V):
         "model_type": "vace",
     }
 
+    def __init__(self, unet_config):
+        super().__init__(unet_config)
+        self.memory_usage_factor = 1.2 * self.memory_usage_factor
+
     def get_model(self, state_dict, prefix="", device=None):
         out = model_base.WAN21_Vace(self, image_to_video=False, device=device)
         return out