EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Fix duplicated handling of GetWeightsByNameReqInput (#2565)

Browse Source
This commit is contained in:
fzyzcjy
2024-12-26 22:49:32 +08:00
committed by GitHub
parent 773951548d
commit 3169e66c23
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/sglang/srt/managers/scheduler.py
View File

@@ -468,9 +468,6 @@ class Scheduler:
self.send_to_tokenizer.send_pyobj( self.send_to_tokenizer.send_pyobj(
UpdateWeightFromDiskReqOutput(success, message) UpdateWeightFromDiskReqOutput(success, message)
) )
elif isinstance(recv_req, GetWeightsByNameReqInput):
parameter = self.get_weights_by_name(recv_req)
self.send_to_tokenizer.send_pyobj(GetWeightsByNameReqOutput(parameter))
elif isinstance(recv_req, InitWeightsUpdateGroupReqInput): elif isinstance(recv_req, InitWeightsUpdateGroupReqInput):
success, message = self.init_weights_update_group(recv_req) success, message = self.init_weights_update_group(recv_req)
self.send_to_tokenizer.send_pyobj( self.send_to_tokenizer.send_pyobj(