[Diffusion] Qwen image support (#82)

Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>

vllm_omni/diffusion/distributed/utils.py

@@ -0,0 +1,11 @@
+# SPDX-License-Identifier: Apache-2.0
+# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
+
+import os
+
+import torch
+
+
+def get_local_device() -> torch.device:
+    """Return the torch device for the current rank."""
+    return torch.device(f"cuda:{os.environ.get('LOCAL_RANK', 0)}")