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_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"