Skip to content

Commit 16145a4

Browse files
committed
Need accelerate
1 parent 4785759 commit 16145a4

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

scripts/compare_vllm.sh

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -140,6 +140,8 @@ if has_aws_creds:
140140
])
141141
142142
commands.extend([
143+
# Install accelerate
144+
"pip install accelerate",
143145
# Run comparison
144146
f'python -m olmocr.train.compare_vllm_checkpoint --model {model_path} --max-tokens {max_tokens} --num-prompts {num_prompts} --prob-threshold {prob_threshold} --seed {seed}'
145147
])

0 commit comments

Comments
 (0)