feat: updated ACE-Step 1.5 version to v1.6.0; add .env to allow user pass env var to ACE-Step 1.5

This commit is contained in:
Xiaohan-Tian
2026-04-18 17:58:07 -07:00
parent cf92498162
commit 5ab16c9812
2 changed files with 26 additions and 4 deletions
+24 -2
View File
@@ -14,6 +14,22 @@ import httpx
ROOT_DIR = Path(__file__).parent.parent
def _load_dotenv(path: Path) -> dict[str, str]:
if not path.exists():
return {}
result: dict[str, str] = {}
for line in path.read_text(encoding="utf-8").splitlines():
line = line.strip()
if not line or line.startswith("#") or "=" not in line:
continue
key, _, value = line.partition("=")
key = key.strip()
value = value.strip().strip("\"'")
if key and value:
result[key] = value
return result
ACESTEP_PORT = 8001
FOUNDATION1_PORT = 8002
HEALTH_POLL_INTERVAL = 2.0
@@ -98,8 +114,14 @@ class ModelManager:
def _launch_acestep(self) -> subprocess.Popen:
venv_python = _python_exe(ROOT_DIR / "ace-step" / ".venv")
cmd = [str(venv_python), "-c", "from acestep.api_server import main; main()"]
env = {**os.environ, "ACESTEP_API_PORT": str(self.acestep_port)}
env.setdefault("ACESTEP_LM_MODEL_PATH", "acestep-5Hz-lm-0.6B")
dotenv_vars = _load_dotenv(ROOT_DIR / ".env")
env = {**os.environ, **dotenv_vars, "ACESTEP_API_PORT": str(self.acestep_port)}
if dotenv_vars:
logger.info("ACE-Step env vars from .env:")
for k, v in dotenv_vars.items():
logger.info(" %s=%s", k, v)
else:
logger.info("No .env file found; no extra env vars applied to ACE-Step.")
logger.info("Launching ACE-Step on port %d: %s", self.acestep_port, " ".join(cmd))
return subprocess.Popen(cmd, cwd=str(ROOT_DIR / "ace-step"), env=env)
+2 -2
View File
@@ -1,8 +1,8 @@
{
"ace-step": {
"url": "https://github.com/ace-step/ACE-Step-1.5.git",
"commit": "2d14b9e070aff945fd8b3dc3491a4386b91702a6",
"description": "Merge PR #793 feat/jetson-docker-support (2026-03-08)",
"commit": "2e786dcfccaecf0666ddf4337d02c89c11c91ef2",
"description": "v0.1.6 XL series 4B DiT decoder (2026-04-03)",
"dir": "ace-step"
},
"foundation1": {