[diffusion] model: Support TurboWan2.2-I2V SLA && add CI test for TurboWan (#16536)

This commit is contained in:
HuangJi
2026-01-12 13:55:38 +08:00
committed by GitHub
parent 38b30c7b56
commit feb39f7768
9 changed files with 247 additions and 25 deletions
@@ -148,6 +148,18 @@ class WanI2V720PConfig(WanI2V480PConfig):
flow_shift: float | None = 5.0
@dataclass
class TurboWanI2V720Config(WanI2V720PConfig):
flow_shift: float | None = 8.0
dmd_denoising_steps: list[int] | None = field(
default_factory=lambda: [996, 932, 852, 608]
)
boundary_ratio: float | None = 0.9
def __post_init__(self) -> None:
self.dit_config.boundary_ratio = self.boundary_ratio
@dataclass
class FastWan2_1_T2V_480P_Config(WanT2V480PConfig):
"""Base configuration for FastWan T2V 1.3B 480P pipeline architecture with DMD"""
@@ -273,6 +273,14 @@ class Wan2_2_I2V_A14B_SamplingParam(Wan2_2_Base_SamplingParams):
)
@dataclass
class Turbo_Wan2_2_I2V_A14B_SamplingParam(Wan2_2_Base_SamplingParams):
guidance_scale: float = 3.5 # high_noise
guidance_scale_2: float = 3.5 # low_noise
num_inference_steps: int = 4
fps: int = 16
# =============================================
# ============= Causal Self-Forcing =============
# =============================================