diff --git a/olmocr/pipeline.py b/olmocr/pipeline.py index 4d3d1cb..f73818c 100644 --- a/olmocr/pipeline.py +++ b/olmocr/pipeline.py @@ -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, + }, }