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

major refactor 2

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-01-22 12:04:29 -05:00
parent 6833d4d820
commit 3a1229afaf
4 changed files with 175 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

12
example_trainer/vllm_patching/patched_gpu_runner.py
View file

@ -295,9 +295,15 @@ 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", ".")
Path(log_dir).mkdir(parents=True, exist_ok=True)
json_path = Path(log_dir) / "vllm_bridge_config.json"
# Allow explicit config path via env var, otherwise use LOGDIR
config_path = os.environ.get("VLLM_BRIDGE_CONFIG_PATH")
if config_path:
json_path = Path(config_path)
json_path.parent.mkdir(parents=True, exist_ok=True)
else:
log_dir = os.environ.get("LOGDIR", ".")
Path(log_dir).mkdir(parents=True, exist_ok=True)
json_path = Path(log_dir) / "vllm_bridge_config.json"
param_mappings = {}
param_names = []