diff --git a/olmocr/pipeline.py b/olmocr/pipeline.py index 167689a..631fb45 100644 --- a/olmocr/pipeline.py +++ b/olmocr/pipeline.py @@ -207,7 +207,7 @@ async def apost(url, json_data): async def process_page(args, worker_id: int, pdf_orig_path: str, pdf_local_path: str, page_num: int) -> PageResult: COMPLETION_URL = f"http://localhost:{BASE_SERVER_PORT}/v1/chat/completions" MAX_RETRIES = args.max_page_retries - TEMPERATURE_BY_ATTEMPT = [0.1, 0.1, 0.2, 0.3, 0.5, 0.8, 0.1, 0.8] + TEMPERATURE_BY_ATTEMPT = [0.1, 0.1, 0.2, 0.3, 0.5, 0.8, 0.9, 1.0] exponential_backoffs = 0 local_anchor_text_len = args.target_anchor_text_len local_image_rotation = 0