[Model] SupportHunyuanImage3 Diffusion Model in GPU

Co-authored-by: skf1999 <13234016272@163.com>
Co-authored-by: Just-it <1161406585@qq.com>
Co-authored-by: Semmer2 <semmer@live.cn>

Signed-off-by: skf1999 <13234016272@163.com>

Author: ElleElleWu

vllm_omni/diffusion/distributed/parallel_state.py

@@ -660,6 +660,7 @@ def initialize_model_parallel(
         backend=backend,
         parallel_mode="data",
     )
+    vllm_parallel_state._DP = _DP
 
     global _CFG
     assert _CFG is None, "classifier_free_guidance group is already initialized"
@@ -677,6 +678,7 @@ def initialize_model_parallel(
         backend=backend,
         parallel_mode="pipeline",
     )
+    vllm_parallel_state._PP = _PP
 
     global _SP
     assert _SP is None, "sequence parallel group is already initialized"

vllm_omni/diffusion/models/hunyuan/__init__.py

@@ -0,0 +1,17 @@
+# SPDX-License-Identifier: Apache-2.0
+# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
+"""hunyuan Image diffusion model components."""
+
+from vllm_omni.diffusion.models.hunyuan.hunyuan_image_3 import (
+    HunyuanImage3Pipeline,
+)
+from vllm_omni.diffusion.models.hunyuan.hunyuan_image_3_models import (
+    HunyuanImage3Model,
+    HunyuanImage3Text2ImagePipeline,
+)
+
+__all__ = [
+    "HunyuanImage3Pipeline",
+    "HunyuanImage3Model",
+    "HunyuanImage3Text2ImagePipeline",
+]