Skip to content

Commit

Permalink
Need those chat templates
Browse files Browse the repository at this point in the history
  • Loading branch information
jakep-allenai committed Mar 5, 2025
1 parent 5611d79 commit c9ecd8e
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions olmocr/bench/scripts/convert_all.sh
Original file line number Diff line number Diff line change
Expand Up @@ -125,18 +125,18 @@ source activate olmocr
# For each model, start server, run benchmark, then stop server

# olmocr_base_temp0_1
start_sglang_server "allenai/olmOCR-7B-0225-preview" --mem-fraction-static 0.7
start_sglang_server "allenai/olmOCR-7B-0225-preview" --chat-template qwen2-vl --mem-fraction-static 0.7
python -m olmocr.bench.convert server:name=olmocr_base_temp0_1:model=allenai/olmOCR-7B-0225-preview:temperature=0.1:prompt_template=fine_tune:response_template=json --repeats 5 --parallel 20
python -m olmocr.bench.convert server:name=olmocr_base_temp0_8:model=allenai/olmOCR-7B-0225-preview:temperature=0.8:prompt_template=fine_tune:response_template=json --repeats 5 --parallel 20
stop_sglang_server

# qwen2_vl_7b
start_sglang_server "Qwen/Qwen2-VL-7B-Instruct" --mem-fraction-static 0.7
start_sglang_server "Qwen/Qwen2-VL-7B-Instruct" --chat-template qwen2-vl --mem-fraction-static 0.7
python -m olmocr.bench.convert server:name=qwen2_vl_7b:model=Qwen/Qwen2-VL-7B-Instruct:temperature=0.1:prompt_template=full:response_template=plain --repeats 5 --parallel 20
stop_sglang_server

# qwen25_vl_7b
start_sglang_server "Qwen/Qwen2.5-VL-7B-Instruct" --mem-fraction-static 0.7
start_sglang_server "Qwen/Qwen2.5-VL-7B-Instruct" --chat-template qwen2-vl --mem-fraction-static 0.7
python -m olmocr.bench.convert server:name=qwen25_vl_7b:model=Qwen/Qwen2.5-VL-7B-Instruct:temperature=0.1:prompt_template=full:response_template=plain --repeats 5 --parallel 20
stop_sglang_server

Expand Down

0 comments on commit c9ecd8e

Please sign in to comment.