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

[Fix] self.worker assignment in TpModelWorker and refactor references (#4788)

Browse Source 
Signed-off-by: Xinyuan Tong <justinning0323@outlook.com>
This commit is contained in:
XinyuanTong
2025-03-27 20:28:38 -07:00
committed by GitHub
parent 04eb6062e4
commit 42a45df043
2 changed files with 4 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1703,18 +1703,12 @@ class Scheduler(
def save_remote_model(self, params): def save_remote_model(self, params):
url = params["url"] url = params["url"]
if isinstance(self.tp_worker, TpModelWorkerClient): worker = self.tp_worker.worker
worker = self.tp_worker.worker
else:
worker = self.tp_worker
worker.model_runner.save_remote_model(url) worker.model_runner.save_remote_model(url)
def save_sharded_model(self, params): def save_sharded_model(self, params):
if isinstance(self.tp_worker, TpModelWorkerClient): worker = self.tp_worker.worker
worker = self.tp_worker.worker
else:
worker = self.tp_worker
worker.model_runner.save_sharded_model( worker.model_runner.save_sharded_model(
path=params["path"], path=params["path"],

2
python/sglang/srt/managers/tp_worker.py
View File

@@ -53,6 +53,8 @@ class TpModelWorker:
req_to_token_pool: Optional[ReqToTokenPool] = None, req_to_token_pool: Optional[ReqToTokenPool] = None,
token_to_kv_pool_allocator: Optional[TokenToKVPoolAllocator] = None, token_to_kv_pool_allocator: Optional[TokenToKVPoolAllocator] = None,
): ):
self.worker = self
# Parse args # Parse args
self.tp_rank = tp_rank self.tp_rank = tp_rank