Skip to content

Commit ed9eb9e

Browse files
Lower wan memory estimation value a bit. (comfyanonymous#9964)
Previous pr reduced the peak memory requirement.
1 parent 9d8935a commit ed9eb9e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

comfy/supported_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -995,7 +995,7 @@ class WAN21_T2V(supported_models_base.BASE):
995995
unet_extra_config = {}
996996
latent_format = latent_formats.Wan21
997997

998-
memory_usage_factor = 1.0
998+
memory_usage_factor = 0.9
999999

10001000
supported_inference_dtypes = [torch.float16, torch.bfloat16, torch.float32]
10011001

@@ -1004,7 +1004,7 @@ class WAN21_T2V(supported_models_base.BASE):
10041004

10051005
def __init__(self, unet_config):
10061006
super().__init__(unet_config)
1007-
self.memory_usage_factor = self.unet_config.get("dim", 2000) / 2000
1007+
self.memory_usage_factor = self.unet_config.get("dim", 2000) / 2222
10081008

10091009
def get_model(self, state_dict, prefix="", device=None):
10101010
out = model_base.WAN21(self, device=device)

0 commit comments

Comments
 (0)