NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-28 17:29:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

health changes

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2025-12-10 16:11:48 -05:00
parent 3b469f2445
commit 78ea8bc3e7
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
example_trainer/grpo.py
View file

@ -909,7 +909,7 @@ def train(config: TrainingConfig):
})
# Check for unexpected vLLM termination
_check_vllm_health()
_check_vllm_process_health()
# === Cleanup ===
save_checkpoint(model, tokenizer, config.save_path, config.training_steps, is_final=True)
@ -989,8 +989,8 @@ def _terminate_vllm_process() -> None:
vllm_process = None
def _check_vllm_health() -> None:
"""Check if vLLM process terminated unexpectedly."""
def _check_vllm_process_health() -> None:
"""Check if vLLM process terminated unexpectedly (legacy mode)."""
global vllm_process
if vllm_process is not None and vllm_process.poll() is not None: