Skip to content

Commit 54719b6

Browse files
committed
Fixed
1 parent c63e97f commit 54719b6

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

olmocr/pipeline.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -602,7 +602,10 @@ async def vllm_server_task(model_name_or_path, args, semaphore):
602602

603603
# Ensure the subprocess is terminated on exit
604604
def _kill_proc():
605-
proc.terminate()
605+
try:
606+
proc.terminate()
607+
except:
608+
logger.info("VLLM Process already terminated")
606609

607610
atexit.register(_kill_proc)
608611

0 commit comments

Comments
 (0)