298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-06-07 03:52:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

Better vace memory estimation. (#7875)

Browse Source
This commit is contained in:
comfyanonymous 2025-04-29 17:42:00 -07:00 committed by GitHub
parent 7ee96455e2
commit dbc726f80c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/ldm/wan/model.py
View File

@ -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)

4
comfy/supported_models.py
View File

@ -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