yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-10-31 10:04:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Better error handling

Browse Source
This commit is contained in:
Jake Poznanski 2024-12-02 23:56:45 +00:00
parent 35502bc6f2
commit fe04db828e
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pdelfin/beakerpipeline.py
View File

@ -517,9 +517,10 @@ async def sglang_server_host(args, semaphore):
logger.warning("SGLang server task ended")
retry += 1
logger.error(f"Ended up restarting the sglang server more than {retry} times, cancelling")
sys.exit(1)
if retry >= MAX_RETRIES:
logger.error(f"Ended up restarting the sglang server more than {retry} times, cancelling")
sys.exit(1)
async def sglang_server_ready():
max_attempts = 300