diff --git a/python/sglang/srt/entrypoints/engine.py b/python/sglang/srt/entrypoints/engine.py index fbd923d91..91e0d695c 100644 --- 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()