Skip to content

Commit 2735d6b

Browse files
ravi03071991Ravi Theja Desetty
andauthored
Fix issue with killing process in sglang (#666)
Co-authored-by: Ravi Theja Desetty <[email protected]>
1 parent ee04e1e commit 2735d6b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lmms_eval/models/srt_api.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
from loguru import logger as eval_logger
1515
from openai import AsyncOpenAI, OpenAI
1616
from PIL import Image
17-
from sglang.srt.utils import kill_child_process
17+
from sglang.srt.utils import kill_process_tree
1818
from sglang.test.test_utils import (
1919
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
2020
popen_launch_server,
@@ -313,7 +313,7 @@ async def _process(request):
313313
pbar.update(1)
314314
else:
315315
asyncio.run(run(requests))
316-
kill_child_process(self.process.pid)
316+
kill_process_tree(self.process.pid)
317317

318318
return res
319319

0 commit comments

Comments
 (0)