Update vllm/executor/ray_gpu_executor.py

Co-authored-by: Cody Yu <hao.yu.cody@gmail.com> Signed-off-by: Rui Qiao <ruisearch42@gmail.com>
vllm-project · Dec 18, 2024 · 0f453a7 · 0f453a7
1 parent 36e9289
commit 0f453a7
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/vllm/executor/ray_gpu_executor.py b/vllm/executor/ray_gpu_executor.py
@@ -174,10 +174,7 @@ def _init_workers_ray(self, placement_group: "PlacementGroup",
         for ip in worker_ips:
             ip_counts[ip] = ip_counts.get(ip, 0) + 1
 
-        worker_to_ip = {
-            worker: ip
-            for worker, ip in zip(self.workers, worker_ips)
-        }
+        worker_to_ip = dict(zip(self.workers, worker_ips))
 
         def sort_by_driver_then_worker_ip(worker):
             """