[diffusion] model: support TurboWan2.1-T2V-1.3B/14B SLA (#15888)
This commit is contained in:
@@ -36,6 +36,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,
|
||||
TurboWanT2V480PConfig,
|
||||
Wan2_2_I2V_A14B_Config,
|
||||
Wan2_2_T2V_A14B_Config,
|
||||
Wan2_2_TI2V_5B_Config,
|
||||
@@ -320,6 +321,13 @@ def _register_configs():
|
||||
],
|
||||
model_detectors=[lambda hf_id: "wanpipeline" in hf_id.lower()],
|
||||
)
|
||||
register_configs(
|
||||
sampling_param_cls=WanT2V_1_3B_SamplingParams,
|
||||
pipeline_config_cls=TurboWanT2V480PConfig,
|
||||
hf_model_paths=[
|
||||
"IPostYellow/TurboWan2.1-T2V-1.3B-Diffusers",
|
||||
],
|
||||
)
|
||||
register_configs(
|
||||
sampling_param_cls=WanT2V_14B_SamplingParams,
|
||||
pipeline_config_cls=WanT2V720PConfig,
|
||||
@@ -327,6 +335,13 @@ def _register_configs():
|
||||
"Wan-AI/Wan2.1-T2V-14B-Diffusers",
|
||||
],
|
||||
)
|
||||
register_configs(
|
||||
sampling_param_cls=WanT2V_14B_SamplingParams,
|
||||
pipeline_config_cls=TurboWanT2V480PConfig,
|
||||
hf_model_paths=[
|
||||
"IPostYellow/TurboWan2.1-T2V-14B-Diffusers",
|
||||
],
|
||||
)
|
||||
register_configs(
|
||||
sampling_param_cls=WanI2V_14B_480P_SamplingParam,
|
||||
pipeline_config_cls=WanI2V480PConfig,
|
||||
|
||||
Reference in New Issue
Block a user