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

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

View File

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