Support profile args in Engine API (#6539)

2025-09-14 01:21:10 -07:00
parent ac964d2e58
commit 4da5533682
1 changed files with 2 additions and 2 deletions
@@ -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()