From ea35a0cc4a35dcdb674af76d8279c084a6aaa181 Mon Sep 17 00:00:00 2001
From: Zeng Chuang <zengchuang3@huawei.com>
Date: Sat, 16 May 2026 20:51:31 +0800
Subject: [PATCH] [Bugfix]update process name for dit stage (#3602)

Signed-off-by: zengchuang <zengchuang3@huawei.com>
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
---
 vllm_omni/diffusion/worker/diffusion_worker.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/vllm_omni/diffusion/worker/diffusion_worker.py b/vllm_omni/diffusion/worker/diffusion_worker.py
index aa0c99cec..e2c4b97c1 100644
--- a/vllm_omni/diffusion/worker/diffusion_worker.py
+++ b/vllm_omni/diffusion/worker/diffusion_worker.py
@@ -796,6 +796,14 @@ class WorkerProc:
         """Worker initialization and execution loops."""
         from vllm_omni.plugins import load_omni_general_plugins
 
+        # Set process title for visibility in nvidia-smi / htop (optional, non-fatal)
+        try:
+            import setproctitle
+
+            setproctitle.setproctitle(f"vLLM-Omni::StageDiffusionProc-{rank}")
+        except ImportError:
+            pass  # setproctitle not installed, skip process title setting
+
         load_omni_general_plugins()
         worker_proc = WorkerProc(
             od_config,