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:
@@ -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
@@ -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": {
|
||||
|
||||
Reference in New Issue
Block a user