Adjust memory estimation code for ACE VAE. (comfyanonymous#7990)

comfyanonymous · web-flow · commit c7c025b8d16f · 2025-05-08T01:22:23.000-04:00
diff --git a/comfy/sd.py b/comfy/sd.py
@@ -442,7 +442,7 @@ def __init__(self, sd=None, device=None, config=None, dtype=None, metadata=None)
             elif "vocoder.backbone.channel_layers.0.0.bias" in sd: #Ace Step Audio
                 self.first_stage_model = comfy.ldm.ace.vae.music_dcae_pipeline.MusicDCAE(source_sample_rate=44100)
                 self.memory_used_encode = lambda shape, dtype: (shape[2] * 300) * model_management.dtype_size(dtype)
-                self.memory_used_decode = lambda shape, dtype: (shape[2] * shape[3] * 72000) * model_management.dtype_size(dtype)
+                self.memory_used_decode = lambda shape, dtype: (shape[2] * shape[3] * 87000) * model_management.dtype_size(dtype)
                 self.latent_channels = 8
                 self.output_channels = 2
                 # self.upscale_ratio = 2048
