[Fix] self.worker assignment in TpModelWorker and refactor references (#4788)
Signed-off-by: Xinyuan Tong <justinning0323@outlook.com>
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user