NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-05-01 17:45:16 +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 eed13670de
commit ca1ec60869
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"

2
example_trainer/vllm_patching/patched_gpu_runner.py
View file

@ -224,7 +224,7 @@ def _create_patched_runner(BaseRunner: type) -> type:
print(f"[vLLM Patch] Note: model.share_memory() not available: {e}")
# Export parameter info to JSON for trainer
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"