fix: pass world_size int to init_distributed_environment instead of vllm_config

This commit is contained in:
2026-02-10 18:58:21 +08:00
parent e75504df72
commit 0765fc9fd3

View File

@@ -85,11 +85,11 @@ class NPUWorker(WorkerBase):
# Initialize distributed (HCCL)
init_distributed_environment(
self.vllm_config,
self.rank,
self.distributed_init_method,
self.local_rank,
"hccl",
world_size=self.parallel_config.world_size,
rank=self.rank,
distributed_init_method=self.distributed_init_method,
local_rank=self.local_rank,
backend="hccl",
)
# Set random seed