Support profile args in Engine API (#6539)
This commit is contained in:
@@ -364,9 +364,9 @@ class Engine(EngineBase):
|
|||||||
loop = asyncio.get_event_loop()
|
loop = asyncio.get_event_loop()
|
||||||
return loop.run_until_complete(self.tokenizer_manager.flush_cache())
|
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 = 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):
|
def stop_profile(self):
|
||||||
loop = asyncio.get_event_loop()
|
loop = asyncio.get_event_loop()
|
||||||
|
|||||||
Reference in New Issue
Block a user