Skip to content

GitLab

Unverified Commit 3169e66c authored by fzyzcjy's avatar fzyzcjy Committed by GitHub
Browse files

Fix duplicated handling of GetWeightsByNameReqInput (#2565)

parent 77395154
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
python/sglang/srt/managers/scheduler.py
View file @ 3169e66c
...@@ -468,9 +468,6 @@ class Scheduler: ...@@ -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(
......
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