[Model] SupportHunyuanImage3 Diffusion Model in GPU

Co-authored-by: ElleElleWu <1608928702@qq.com>
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: Semmer2 <semmer@live.cn>

Author: ElleElleWu

pyproject.toml

@@ -184,6 +184,7 @@ extend-ignore-identifiers-re = [
     ".*arange",
     ".*MoBA",
     ".*temperal_downsample",
+    ".*_nd",
     ".*nothink.*",
     ".*NOTHINK.*",
     ".*nin.*",

vllm_omni/diffusion/distributed/parallel_state.py

@@ -757,6 +757,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"
@@ -774,6 +775,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_image_3/__init__.py

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