Fix stop_profile does not wait for finishing (#4741)

2025-05-18 08:06:15 +08:00
parent e3b8a72291
commit 01d2838c0f
4 changed files with 19 additions and 11 deletions
--- a/python/sglang/srt/entrypoints/engine.py
+++ b/python/sglang/srt/entrypoints/engine.py
@@ -321,7 +321,8 @@ class Engine(EngineBase):
        loop.run_until_complete(self.tokenizer_manager.start_profile())

    def stop_profile(self):
-        self.tokenizer_manager.stop_profile()
+        loop = asyncio.get_event_loop()
+        loop.run_until_complete(self.tokenizer_manager.stop_profile())

    def get_server_info(self):
        loop = asyncio.get_event_loop()