6838533998
Signed-off-by: lishunyang <lishunyang12@163.com> Signed-off-by: 汪志鹏 <wangzhipeng628@gmail.com> Co-authored-by: 汪志鹏 <wangzhipeng628@gmail.com> Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
26 lines
930 B
Python
26 lines
930 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
"""HunyuanVideo-1.5 diffusion model components (T2V and I2V)."""
|
|
|
|
from vllm_omni.diffusion.models.hunyuan_video.hunyuan_video_15_transformer import (
|
|
HunyuanVideo15Transformer3DModel,
|
|
)
|
|
from vllm_omni.diffusion.models.hunyuan_video.pipeline_hunyuan_video_1_5 import (
|
|
HunyuanVideo15Pipeline,
|
|
get_hunyuan_video_15_post_process_func,
|
|
)
|
|
from vllm_omni.diffusion.models.hunyuan_video.pipeline_hunyuan_video_1_5_i2v import (
|
|
HunyuanVideo15I2VPipeline,
|
|
get_hunyuan_video_15_i2v_post_process_func,
|
|
get_hunyuan_video_15_i2v_pre_process_func,
|
|
)
|
|
|
|
__all__ = [
|
|
"HunyuanVideo15Pipeline",
|
|
"HunyuanVideo15I2VPipeline",
|
|
"HunyuanVideo15Transformer3DModel",
|
|
"get_hunyuan_video_15_post_process_func",
|
|
"get_hunyuan_video_15_i2v_post_process_func",
|
|
"get_hunyuan_video_15_i2v_pre_process_func",
|
|
]
|