Skip to content

Commit 7689eb2

Browse files
Run two engines in single GPU script.
1 parent 8b2539b commit 7689eb2

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

scripts/train/debug/single_gpu_on_beaker.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@ uv run python mason.py \
4747
--with_tracking \
4848
--num_epochs 1 \
4949
--num_learners_per_node 1 \
50+
--vllm_num_engines 2 \
5051
--vllm_tensor_parallel_size 1 \
5152
--beta 0.01 \
5253
--seed 3 \

0 commit comments

Comments
 (0)