yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-11-25 14:52:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Trying fix for transformers benchmark

Browse Source
This commit is contained in:
Jake Poznanski 2025-08-04 19:50:05 +00:00
parent 8712534e81
commit 7c098955a9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/run_transformers_benchmark.sh
View File

@ -92,7 +92,7 @@ commands.extend([
"cd olmOCR-bench && git lfs pull && cd ..", "cd olmOCR-bench && git lfs pull && cd ..",
"pip install accelerate", "pip install accelerate",
"pip install flash-attn==2.8.0.post2 --no-build-isolation", "pip install flash-attn==2.8.0.post2 --no-build-isolation",
"python -m olmocr.bench.convert transformers:target_longest_image_dim=1288:prompt_template=yaml:response_template=yaml --dir ./olmOCR-bench/bench_data", "PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True python -m olmocr.bench.convert transformers:target_longest_image_dim=1288:prompt_template=yaml:response_template=yaml --dir ./olmOCR-bench/bench_data",
"python -m olmocr.bench.benchmark --dir ./olmOCR-bench/bench_data" "python -m olmocr.bench.benchmark --dir ./olmOCR-bench/bench_data"
]) ])