From 2b3a1c820049664dfd8bda14714defaf9298e4f5 Mon Sep 17 00:00:00 2001 From: hsliuustc0106 Date: Sat, 13 Jun 2026 14:43:32 +0800 Subject: [PATCH] Format diffusion RPC guard Signed-off-by: hsliuustc0106 --- vllm_omni/diffusion/worker/diffusion_worker.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/vllm_omni/diffusion/worker/diffusion_worker.py b/vllm_omni/diffusion/worker/diffusion_worker.py index dbb461400..fc46ae890 100644 --- a/vllm_omni/diffusion/worker/diffusion_worker.py +++ b/vllm_omni/diffusion/worker/diffusion_worker.py @@ -763,9 +763,7 @@ class WorkerProc: collect_rank_status = rpc_request.get("collect_rank_status", False) if collect_rank_status and not exec_all_ranks: - raise ValueError( - "collect_rank_status requires exec_all_ranks=True so all ranks enter the status gather" - ) + raise ValueError("collect_rank_status requires exec_all_ranks=True so all ranks enter the status gather") should_execute = exec_all_ranks or output_rank is None or output_rank == self.gpu_id should_reply = (output_rank is None or output_rank == self.gpu_id) and self.result_mq is not None