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

major refactor 4

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-01-22 13:39:34 -05:00
parent 1b1d8a5909
commit 59f337e34f
4 changed files with 18 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

6
example_trainer/scripts/run_concurrent_tests.sh
View file

@ -139,10 +139,11 @@ echo "[3/6] Starting LoRA GSM8k environment..."
python -u environments/gsm8k_server.py serve \
--env.tokenizer_name "$MODEL" \
--env.use_wandb=False \
--env.rollout_server_url "http://localhost:${LORA_GSM8K_PORT}" \
--openai.model_name "$MODEL" \
--openai.base_url "http://localhost:${LORA_VLLM_PORT}/v1" \
--openai.server_type vllm \
--server.port $LORA_GSM8K_PORT \
--slurm false \
> "${LOG_DIR}/lora_gsm8k.log" 2>&1 &
LORA_GSM8K_PID=$!
echo " PID: $LORA_GSM8K_PID"
@ -152,10 +153,11 @@ echo "[4/6] Starting Single-Copy GSM8k environment..."
python -u environments/gsm8k_server.py serve \
--env.tokenizer_name "$MODEL" \
--env.use_wandb=False \
--env.rollout_server_url "http://localhost:${SINGLE_COPY_GSM8K_PORT}" \
--openai.model_name "$MODEL" \
--openai.base_url "http://localhost:${SINGLE_COPY_VLLM_PORT}/v1" \
--openai.server_type vllm \
--server.port $SINGLE_COPY_GSM8K_PORT \
--slurm false \
> "${LOG_DIR}/single_copy_gsm8k.log" 2>&1 &
SINGLE_COPY_GSM8K_PID=$!
echo " PID: $SINGLE_COPY_GSM8K_PID"