Support profile args in Engine API (#6539)

2025-09-14 16:21:10 +08:00
parent ac964d2e58
commit 4da5533682
1 changed files with 2 additions and 2 deletions
--- a/python/sglang/srt/entrypoints/engine.py
+++ b/python/sglang/srt/entrypoints/engine.py
@@ -364,9 +364,9 @@ class Engine(EngineBase):
        loop = asyncio.get_event_loop()
        return loop.run_until_complete(self.tokenizer_manager.flush_cache())

-    def start_profile(self):
+    def start_profile(self, **kwargs):
        loop = asyncio.get_event_loop()
-        loop.run_until_complete(self.tokenizer_manager.start_profile())
+        loop.run_until_complete(self.tokenizer_manager.start_profile(**kwargs))

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