zzh/vllm-npu-plugin
1
0
Fork 0
mirror of https://github.com/handsomezhuzhu/vllm-npu-plugin.git synced 2026-02-20 19:50:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: pass world_size int to init_distributed_environment instead of vllm_config

Browse Source
This commit is contained in:
handsomezhuzhu
2026-02-10 18:58:21 +08:00
parent e75504df72
commit 0765fc9fd3
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
vllm_npu/worker/worker_v1.py
View File

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