optimize lowvram for music

2026-09-02 02:51:17 +02:00 · 2026-04-12 18:17:08 +08:00
parent 5361b45fba
commit 7bf7b0aefc
1 changed files with 5 additions and 1 deletions
@@ -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<float> 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();
    }