[Diffusion] Fix FastWan2.1 default 480p resolution (#28733)

This commit is contained in:
Xiaoyu Zhang
2026-06-22 18:43:24 +08:00
committed by GitHub
parent 4e1d25117b
commit 0c9e775f2c
2 changed files with 8 additions and 1 deletions
@@ -171,7 +171,7 @@ class FastWanT2V480PConfig(WanT2V_1_3B_SamplingParams):
# dmd_denoising_steps: list[int] | None = field(default_factory=lambda: [1000, 757, 522])
num_inference_steps: int = 3
num_frames: int = 61
height: int = 448
height: int = 480
width: int = 832
fps: int = 16
@@ -24,6 +24,7 @@ from sglang.multimodal_gen.configs.sample.sampling_params import (
)
from sglang.multimodal_gen.configs.sample.teacache import TeaCacheParams
from sglang.multimodal_gen.configs.sample.wan import (
FastWanT2V480PConfig,
WanI2V_14B_480P_SamplingParam,
WanI2V_14B_720P_SamplingParam,
WanT2V_1_3B_SamplingParams,
@@ -102,6 +103,12 @@ class TestSamplingParamsSubclass(unittest.TestCase):
with self.assertRaises(AssertionError):
DiffusersGenericSamplingParams(num_frames=0)
def test_fastwan_480p_default_resolution_is_supported(self):
params = FastWanT2V480PConfig()
self.assertEqual((params.width, params.height), (832, 480))
self.assertIn((params.width, params.height), params.supported_resolutions)
def test_output_file_name_supports_callable_teacache_params(self):
def coefficients_callback(_: TeaCacheParams) -> list[float]:
return [1.0, 2.0, 3.0, 4.0, 5.0]