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

Trying beam search idea

Browse Source
This commit is contained in:
Jake Poznanski 2025-10-07 04:00:39 +00:00
parent c89787183a
commit 44f6b9f0de
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
olmocr/pipeline.py
View File

@ -145,6 +145,10 @@ async def build_page_query(local_pdf_path: str, page: int, target_longest_image_
],
"max_tokens": MAX_TOKENS,
"temperature": 0.0,
"extra_body": {
"use_beam_search": True,
"best_of": 5,
},
}