Skip to content

Commit

Permalink
No keep alive connection to try to resolve sglang livelock
Browse files Browse the repository at this point in the history
  • Loading branch information
jakep-allenai committed Nov 21, 2024
1 parent 53a5104 commit 9112d81
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pdelfin/beakerpipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -326,7 +326,7 @@ async def worker(args, work_queue: S3WorkQueue, semaphore, worker_id):
await tracker.clear_work(worker_id)

try:
async with httpx.AsyncClient(timeout=600, limits=httpx.Limits(max_connections=1000)) as session:
async with httpx.AsyncClient(timeout=600, limits=httpx.Limits(max_keepalive_connections=0, max_connections=1000)) as session:
async with asyncio.TaskGroup() as tg:
dolma_tasks = [tg.create_task(process_pdf(args, session, worker_id, pdf)) for pdf in work_item.s3_work_paths]
logger.info(f"Created all tasks for {work_item.hash}")
Expand Down

0 comments on commit 9112d81

Please sign in to comment.