From 14b9b2dc8fbbf9964be2b85429b1fe812f7a9f34 Mon Sep 17 00:00:00 2001 From: Jake Poznanski Date: Fri, 27 Jun 2025 16:23:31 +0000 Subject: [PATCH] Fix for rocm vllm --- olmocr/pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/olmocr/pipeline.py b/olmocr/pipeline.py index f056cd6..36689c3 100644 --- a/olmocr/pipeline.py +++ b/olmocr/pipeline.py @@ -623,7 +623,7 @@ async def vllm_server_task(model_name_or_path, args, semaphore): if match: last_running_req = int(match.group(1)) - match = re.search(r"Waiting: (\d+)", line) + match = re.search(r'(?:Waiting|Pending):\s*(\d+)', line) if match: last_queue_req = int(match.group(1)) logger.info(f"vllm running req: {last_running_req} queue req: {last_queue_req}")