42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
from .patch_diffusers import patch_wan_rms_norm
|
|
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 .wan2_2_transformer import WanTransformer3DModel
|
|
|
|
__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",
|
|
"WanTransformer3DModel",
|
|
]
|
|
|
|
patch_wan_rms_norm()
|