From ceb50b2cbfb166b84786d12d313a624273590fab Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Fri, 27 Dec 2024 07:30:09 -0500
Subject: [PATCH] Closer memory estimation for pixart models.

---
 comfy/supported_models.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/comfy/supported_models.py b/comfy/supported_models.py
index a5f38b5e..512515ae 100644
--- a/comfy/supported_models.py
+++ b/comfy/supported_models.py
@@ -608,6 +608,8 @@ class PixArtAlpha(supported_models_base.BASE):
     unet_extra_config = {}
     latent_format = latent_formats.SD15
 
+    memory_usage_factor = 0.5
+
     vae_key_prefix = ["vae."]
     text_encoder_key_prefix = ["text_encoders."]