major refactor 4

This commit is contained in:
Jai Suphavadeeprasit 2026-01-22 13:39:41 -05:00
parent 40a0c47fc3
commit 2e2fdc2058

View file

@ -75,10 +75,11 @@ echo "[2/4] Starting GSM8k environment..."
python -u environments/gsm8k_server.py serve \
--env.tokenizer_name "$MODEL" \
--env.use_wandb=False \
--env.rollout_server_url "http://localhost:${GSM8K_PORT}" \
--openai.model_name "$MODEL" \
--openai.base_url "http://localhost:${VLLM_PORT}/v1" \
--openai.server_type vllm \
--server.port $GSM8K_PORT \
--slurm false \
> "${LOG_DIR}/gsm8k.log" 2>&1 &
echo "Waiting for GSM8k (10s)..."