From 0765fc9fd319fe752ec0242d2b4bb26a55a737f5 Mon Sep 17 00:00:00 2001
From: handsomezhuzhu <2658601135@qq.com>
Date: Tue, 10 Feb 2026 18:58:21 +0800
Subject: [PATCH] fix: pass world_size int to init_distributed_environment
 instead of vllm_config

---
 vllm_npu/worker/worker_v1.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/vllm_npu/worker/worker_v1.py b/vllm_npu/worker/worker_v1.py
index 82810e3..5e33b0d 100644
--- a/vllm_npu/worker/worker_v1.py
+++ b/vllm_npu/worker/worker_v1.py
@@ -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