From f378f80430b1cac2a3148dd86ad3bd68b6bc73c8 Mon Sep 17 00:00:00 2001
From: Samit <285365963@qq.com>
Date: Mon, 1 Dec 2025 15:17:56 +0800
Subject: [PATCH] [Bugfix] Fix redundant shm broadcast warnings in diffusion
 workers (#133)

Signed-off-by: samithuang <285365963@qq.com>
---
 vllm_omni/diffusion/worker/gpu_worker.py     | 2 +-
 vllm_omni/diffusion/worker/npu/npu_worker.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/vllm_omni/diffusion/worker/gpu_worker.py b/vllm_omni/diffusion/worker/gpu_worker.py
index 84ede7ffd..9428acb95 100644
--- a/vllm_omni/diffusion/worker/gpu_worker.py
+++ b/vllm_omni/diffusion/worker/gpu_worker.py
@@ -129,7 +129,7 @@ class WorkerProc:
         """
         Receive requests from broadcast queue
         """
-        return self.mq.dequeue()
+        return self.mq.dequeue(indefinite=True)
 
     # TODO: queueing, cancellation
     def worker_busy_loop(self) -> None:
diff --git a/vllm_omni/diffusion/worker/npu/npu_worker.py b/vllm_omni/diffusion/worker/npu/npu_worker.py
index 4baf20603..d35d187b4 100644
--- a/vllm_omni/diffusion/worker/npu/npu_worker.py
+++ b/vllm_omni/diffusion/worker/npu/npu_worker.py
@@ -129,7 +129,7 @@ class NPUWorkerProc:
         """
         Receive requests from broadcast queue
         """
-        return self.mq.dequeue()
+        return self.mq.dequeue(indefinite=True)
 
     # TODO: queueing, cancellation
     def worker_busy_loop(self) -> None: