Initialize 5Hz LM on Space startup (ACESTEP_INIT_LLM env, default true)

README.md

@@ -16,4 +16,4 @@ models:
 
 Lyric-controllable, open-source text-to-music. Runs as a Docker Space with GPU.
 
-Models are downloaded from the Hub on first run. **Space Settings → Variables:** set `ACESTEP_CHECKPOINT_REPO` to the checkpoint repo (default: `ACE-Step/Ace-Step1.5`). Select **GPU** (e.g. T4 or A10G) in Space Settings.
+Models are downloaded from the Hub on first run. **Space Settings → Variables:** `ACESTEP_CHECKPOINT_REPO` = checkpoint repo (default: `ACE-Step/Ace-Step1.5`); `ACESTEP_INIT_LLM` = `true` to initialize 5Hz LM on startup (default), `false` to skip and save VRAM. Select **GPU** (e.g. T4 or A10G) in Space Settings.

app.py

@@ -22,7 +22,7 @@ if _REPO_ROOT not in sys.path:
 
 # Override argv for Space: bind all interfaces, port 7860, init service
 os.chdir(_REPO_ROOT)
-sys.argv = [
+argv = [
     sys.argv[0],
     "--server-name", "0.0.0.0",
     "--port", "7860",
@@ -30,6 +30,13 @@ sys.argv = [
     "--config_path", "acestep-v15-turbo",
     "--download-source", "huggingface",
 ]
+# Initialize 5Hz LM by default in Space (set ACESTEP_INIT_LLM=false in Settings to disable and save VRAM)
+_init_lm = os.environ.get("ACESTEP_INIT_LLM", "").strip().lower()
+if _init_lm in ("", "1", "true", "yes", "y", "on", "auto"):
+    argv.extend(["--init_llm", "true"])
+elif _init_lm in ("0", "false", "no", "n", "off"):
+    argv.extend(["--init_llm", "false"])
+sys.argv = argv
 
 from acestep.acestep_v15_pipeline import main