[diffusion] model: Support TurboWan2.2-I2V SLA && add CI test for TurboWan (#16536)
This commit is contained in:
@@ -49,6 +49,7 @@ from sglang.multimodal_gen.configs.pipeline_configs.qwen_image import (
|
||||
from sglang.multimodal_gen.configs.pipeline_configs.wan import (
|
||||
FastWan2_1_T2V_480P_Config,
|
||||
FastWan2_2_TI2V_5B_Config,
|
||||
TurboWanI2V720Config,
|
||||
TurboWanT2V480PConfig,
|
||||
Wan2_2_I2V_A14B_Config,
|
||||
Wan2_2_T2V_A14B_Config,
|
||||
@@ -67,6 +68,7 @@ from sglang.multimodal_gen.configs.sample.qwenimage import (
|
||||
)
|
||||
from sglang.multimodal_gen.configs.sample.wan import (
|
||||
FastWanT2V480PConfig,
|
||||
Turbo_Wan2_2_I2V_A14B_SamplingParam,
|
||||
Wan2_1_Fun_1_3B_InP_SamplingParams,
|
||||
Wan2_2_I2V_A14B_SamplingParam,
|
||||
Wan2_2_T2V_A14B_SamplingParam,
|
||||
@@ -426,6 +428,7 @@ def _register_configs():
|
||||
pipeline_config_cls=TurboWanT2V480PConfig,
|
||||
hf_model_paths=[
|
||||
"IPostYellow/TurboWan2.1-T2V-14B-Diffusers",
|
||||
"IPostYellow/TurboWan2.1-T2V-14B-720P-Diffusers",
|
||||
],
|
||||
)
|
||||
register_configs(
|
||||
@@ -443,6 +446,13 @@ def _register_configs():
|
||||
"Wan-AI/Wan2.1-I2V-14B-720P-Diffusers",
|
||||
],
|
||||
)
|
||||
register_configs(
|
||||
sampling_param_cls=Turbo_Wan2_2_I2V_A14B_SamplingParam,
|
||||
pipeline_config_cls=TurboWanI2V720Config,
|
||||
hf_model_paths=[
|
||||
"IPostYellow/TurboWan2.2-I2V-A14B-Diffusers",
|
||||
],
|
||||
)
|
||||
register_configs(
|
||||
sampling_param_cls=Wan2_1_Fun_1_3B_InP_SamplingParams,
|
||||
pipeline_config_cls=WanI2V480PConfig,
|
||||
|
||||
Reference in New Issue
Block a user