Skip to content

Commit 271c9c5

Browse files
Better mem estimation for the LTXV 13B model. (#7963)
1 parent a4e6797 commit 271c9c5

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

comfy/supported_models.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -785,6 +785,10 @@ class LTXV(supported_models_base.BASE):
785785
vae_key_prefix = ["vae."]
786786
text_encoder_key_prefix = ["text_encoders."]
787787

788+
def __init__(self, unet_config):
789+
super().__init__(unet_config)
790+
self.memory_usage_factor = (unet_config.get("cross_attention_dim", 2048) / 2048) * 5.5
791+
788792
def get_model(self, state_dict, prefix="", device=None):
789793
out = model_base.LTXV(self, device=device)
790794
return out

0 commit comments

Comments
 (0)