NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

improve default

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-01-13 00:40:11 -05:00
parent 9df62a8f64
commit 64cf8e3f82
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
example_trainer/vllm_api_server.py
View file

@ -626,7 +626,7 @@ async def init_app(args: Namespace, llm_engine: AsyncLLM | None = None) -> FastA
def _export_state_dict_info(args: Namespace) -> None:
"""Export basic model info to JSON for trainer (backup if patches don't run)."""
log_dir = os.environ.get("LOGDIR", "/tmp/atropos_bridge")
log_dir = os.environ.get("LOGDIR", ".")
Path(log_dir).mkdir(parents=True, exist_ok=True)
json_path = Path(log_dir) / "vllm_bridge_config.json"