25 lines
930 B
Python
25 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",
|
||
|
|
]
|