[Bugfix]update process name for dit stage (#3602)

Signed-off-by: zengchuang <zengchuang3@huawei.com>
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
This commit is contained in:
Zeng ChuangandHongsheng Liu authored and GitHub committed 2026-05-16 20:51:31 +08:00
1 parent 0f4853ff86
commit ea35a0cc4a
1 file changed
+8
@@ -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,