From 7bf7b0aefc261deb388d6c26bbfa268672deef67 Mon Sep 17 00:00:00 2001 From: Concedo <39025047+LostRuins@users.noreply.github.com> Date: Sun, 12 Apr 2026 18:17:08 +0800 Subject: [PATCH] optimize lowvram for music --- otherarch/acestep/dit-vae.cpp | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/otherarch/acestep/dit-vae.cpp b/otherarch/acestep/dit-vae.cpp index ff84e89ec..c10088a47 100644 --- a/otherarch/acestep/dit-vae.cpp +++ b/otherarch/acestep/dit-vae.cpp @@ -549,6 +549,11 @@ std::string acestep_generate_audio(const music_generation_inputs inputs) } } + if(acestep_dit_lowvram) + { + unload_acestep_dit_others(); + } + // DiT Generate std::vector output(batch_n * Oc * T); @@ -626,7 +631,6 @@ std::string acestep_generate_audio(const music_generation_inputs inputs) if(acestep_dit_lowvram) { - unload_acestep_dit_others(); unload_acestep_vae_dec(); }