Format diffusion RPC guard

Signed-off-by: hsliuustc0106 <liuhongsheng4@huawei.com>
This commit is contained in:
hsliuustc0106 committed 2026-06-13 14:43:32 +08:00
1 parent 08b2f694f3
commit 2b3a1c8200
1 file changed
+1 -3
@@ -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