[BUG] fix stop_profile crash (#6431)
This commit is contained in:
@@ -807,6 +807,7 @@ class TokenizerManager:
|
||||
return await self._execute_profile(req)
|
||||
|
||||
async def stop_profile(self):
|
||||
self.auto_create_handle_loop()
|
||||
req = ProfileReq(type=ProfileReqType.STOP_PROFILE)
|
||||
return await self._execute_profile(req)
|
||||
|
||||
@@ -816,11 +817,6 @@ class TokenizerManager:
|
||||
raise RuntimeError(result.message)
|
||||
return result
|
||||
|
||||
def stop_profile(self):
|
||||
self.auto_create_handle_loop()
|
||||
req = ProfileReq(type=ProfileReqType.STOP_PROFILE)
|
||||
self.send_to_scheduler.send_pyobj(req)
|
||||
|
||||
async def start_expert_distribution_record(self):
|
||||
self.auto_create_handle_loop()
|
||||
await self.expert_distribution_communicator(ExpertDistributionReq.START_RECORD)
|
||||
|
||||
Reference in New Issue
Block a user