mirror of
https://github.com/handsomezhuzhu/vllm-npu-plugin.git
synced 2026-02-20 11:42:30 +00:00
fix: pass world_size int to init_distributed_environment instead of vllm_config
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user