yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-11-07 22:18:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Need accelerate

Browse Source
This commit is contained in:
Jake Poznanski 2025-07-16 18:51:37 +00:00
parent 4785759ba1
commit 16145a4b32
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/compare_vllm.sh
View File

@ -140,6 +140,8 @@ if has_aws_creds:
]) ])
commands.extend([ commands.extend([
# Install accelerate
"pip install accelerate",
# Run comparison # Run comparison
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}' 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}'
]) ])