fix init error for MessageQueue when n_local_reader is zero (#11768)

2025-01-07 14:12:48 +08:00 · 2025-01-07 14:12:48 +08:00 · e512f76a89
commit e512f76a89
parent 898cdf033e
1 changed files with 2 additions and 1 deletions
--- a/vllm/distributed/device_communicators/shm_broadcast.py
+++ b/vllm/distributed/device_communicators/shm_broadcast.py
@ -247,7 +247,8 @@ class MessageQueue:
        self.handle = Handle(
            connect_ip=connect_ip,
            local_reader_ranks=local_reader_ranks,
-            buffer_handle=self.buffer.handle(),
+            buffer_handle=self.buffer.handle()
+            if self.buffer is not None else None,
            local_subscribe_port=local_subscribe_port,
            remote_subscribe_port=remote_subscribe_port,
        )