Skip to content

GitLab

"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "5fcee4a4471d32d3a5959e55805303a7ec7a801e"
Unverified Commit b06215da authored by Yi Zhang's avatar Yi Zhang Committed by GitHub
Browse files

[BUG] fix stop_profile crash (#6431)

parent 7adf245b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 5 deletions
python/sglang/srt/managers/tokenizer_manager.py
View file @ b06215da
...@@ -807,6 +807,7 @@ class TokenizerManager: ...@@ -807,6 +807,7 @@ class TokenizerManager:
return await self._execute_profile(req) return await self._execute_profile(req)
async def stop_profile(self): async def stop_profile(self):
self.auto_create_handle_loop()
req = ProfileReq(type=ProfileReqType.STOP_PROFILE) req = ProfileReq(type=ProfileReqType.STOP_PROFILE)
return await self._execute_profile(req) return await self._execute_profile(req)
...@@ -816,11 +817,6 @@ class TokenizerManager: ...@@ -816,11 +817,6 @@ class TokenizerManager:
raise RuntimeError(result.message) raise RuntimeError(result.message)
return result 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): async def start_expert_distribution_record(self):
self.auto_create_handle_loop() self.auto_create_handle_loop()
await self.expert_distribution_communicator(ExpertDistributionReq.START_RECORD) await self.expert_distribution_communicator(ExpertDistributionReq.START_RECORD)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment