wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-12-18 02:23:06 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Lower wan memory estimation value a bit. (#9964)

Browse Source 
Previous pr reduced the peak memory requirement.
This commit is contained in:
comfyanonymous 2025-09-20 19:09:35 -07:00 committed by GitHub
parent 7be2b49b6b
commit d1d9eb94b1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/supported_models.py
View File

@ -995,7 +995,7 @@ class WAN21_T2V(supported_models_base.BASE):
unet_extra_config = {} unet_extra_config = {}
latent_format = latent_formats.Wan21 latent_format = latent_formats.Wan21
memory_usage_factor = 1.0 memory_usage_factor = 0.9
supported_inference_dtypes = [torch.float16, torch.bfloat16, torch.float32] supported_inference_dtypes = [torch.float16, torch.bfloat16, torch.float32]
@ -1004,7 +1004,7 @@ class WAN21_T2V(supported_models_base.BASE):
def __init__(self, unet_config): def __init__(self, unet_config):
super().__init__(unet_config) super().__init__(unet_config)
self.memory_usage_factor = self.unet_config.get("dim", 2000) / 2000 self.memory_usage_factor = self.unet_config.get("dim", 2000) / 2222
def get_model(self, state_dict, prefix="", device=None): def get_model(self, state_dict, prefix="", device=None):
out = model_base.WAN21(self, device=device) out = model_base.WAN21(self, device=device)