Closer memory estimation for pixart models.

comfyanonymous · Dec 27, 2024 · ceb50b2 · ceb50b2
1 parent 160ca08
commit ceb50b2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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."]