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
|
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
|
ACESTEP_PORT = 8001
|
||||||
FOUNDATION1_PORT = 8002
|
FOUNDATION1_PORT = 8002
|
||||||
HEALTH_POLL_INTERVAL = 2.0
|
HEALTH_POLL_INTERVAL = 2.0
|
||||||
@@ -98,8 +114,14 @@ class ModelManager:
|
|||||||
def _launch_acestep(self) -> subprocess.Popen:
|
def _launch_acestep(self) -> subprocess.Popen:
|
||||||
venv_python = _python_exe(ROOT_DIR / "ace-step" / ".venv")
|
venv_python = _python_exe(ROOT_DIR / "ace-step" / ".venv")
|
||||||
cmd = [str(venv_python), "-c", "from acestep.api_server import main; main()"]
|
cmd = [str(venv_python), "-c", "from acestep.api_server import main; main()"]
|
||||||
env = {**os.environ, "ACESTEP_API_PORT": str(self.acestep_port)}
|
dotenv_vars = _load_dotenv(ROOT_DIR / ".env")
|
||||||
env.setdefault("ACESTEP_LM_MODEL_PATH", "acestep-5Hz-lm-0.6B")
|
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))
|
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)
|
return subprocess.Popen(cmd, cwd=str(ROOT_DIR / "ace-step"), env=env)
|
||||||
|
|
||||||
|
|||||||
+2
-2
@@ -1,8 +1,8 @@
|
|||||||
{
|
{
|
||||||
"ace-step": {
|
"ace-step": {
|
||||||
"url": "https://github.com/ace-step/ACE-Step-1.5.git",
|
"url": "https://github.com/ace-step/ACE-Step-1.5.git",
|
||||||
"commit": "2d14b9e070aff945fd8b3dc3491a4386b91702a6",
|
"commit": "2e786dcfccaecf0666ddf4337d02c89c11c91ef2",
|
||||||
"description": "Merge PR #793 feat/jetson-docker-support (2026-03-08)",
|
"description": "v0.1.6 XL series 4B DiT decoder (2026-04-03)",
|
||||||
"dir": "ace-step"
|
"dir": "ace-step"
|
||||||
},
|
},
|
||||||
"foundation1": {
|
"foundation1": {
|
||||||
|
|||||||
Reference in New Issue
Block a user