458f402323
Signed-off-by: Binh Tang <tangbinhna@gmail.com> Signed-off-by: Binh Tang <binht@netflix.com> Signed-off-by: Didan Deng <33117903+wtomin@users.noreply.github.com> Co-authored-by: Binh Tang <binht@netflix.com> Co-authored-by: Didan Deng <33117903+wtomin@users.noreply.github.com>
50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
from .pipeline_wan2_2 import (
|
|
Wan22Pipeline,
|
|
create_transformer_from_config,
|
|
get_wan22_post_process_func,
|
|
get_wan22_pre_process_func,
|
|
load_transformer_config,
|
|
retrieve_latents,
|
|
)
|
|
from .pipeline_wan2_2_i2v import (
|
|
Wan22I2VPipeline,
|
|
get_wan22_i2v_post_process_func,
|
|
get_wan22_i2v_pre_process_func,
|
|
)
|
|
from .pipeline_wan2_2_ti2v import (
|
|
Wan22TI2VPipeline,
|
|
get_wan22_ti2v_post_process_func,
|
|
get_wan22_ti2v_pre_process_func,
|
|
)
|
|
from .pipeline_wan2_2_vace import (
|
|
Wan22VACEPipeline,
|
|
get_wan22_vace_post_process_func,
|
|
get_wan22_vace_pre_process_func,
|
|
)
|
|
from .wan2_2_transformer import WanTransformer3DModel
|
|
from .wan2_2_vace_transformer import VaceWanTransformerBlock, WanVACETransformer3DModel
|
|
|
|
__all__ = [
|
|
"Wan22Pipeline",
|
|
"get_wan22_post_process_func",
|
|
"get_wan22_pre_process_func",
|
|
"retrieve_latents",
|
|
"load_transformer_config",
|
|
"create_transformer_from_config",
|
|
"Wan22I2VPipeline",
|
|
"get_wan22_i2v_post_process_func",
|
|
"get_wan22_i2v_pre_process_func",
|
|
"Wan22TI2VPipeline",
|
|
"get_wan22_ti2v_post_process_func",
|
|
"get_wan22_ti2v_pre_process_func",
|
|
"Wan22VACEPipeline",
|
|
"get_wan22_vace_post_process_func",
|
|
"get_wan22_vace_pre_process_func",
|
|
"WanTransformer3DModel",
|
|
"VaceWanTransformerBlock",
|
|
"WanVACETransformer3DModel",
|
|
]
|