diff --git a/vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py b/vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py index 3a8400447..c2777b393 100644 --- a/vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py +++ b/vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py @@ -2323,15 +2323,15 @@ class NixlConnectorWorker: # Get descs ids for the layer. layer_local_desc_ids = self._get_block_descs_ids( - dst_engine_id, + self.engine_id, layer_local_block_ids, layer_idx, + block_size_ratio=block_size_ratio, ) layer_remote_desc_ids = self._get_block_descs_ids( - self.engine_id, + dst_engine_id, layer_remote_block_ids, layer_idx, - block_size_ratio=block_size_ratio, ) local_descs_list.append(layer_local_desc_ids)