diff --git a/pdelfin/beakerpipeline.py b/pdelfin/beakerpipeline.py index bc40411..77f23ed 100644 --- a/pdelfin/beakerpipeline.py +++ b/pdelfin/beakerpipeline.py @@ -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