[diffusion] feat: support resolution check for video model (#14881)

Co-authored-by: Brain97 <Brain97@users.noreply.github.com>
Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
blahblah
2025-12-14 17:50:13 +08:00
committed by GitHub
co-authored by Brain97 Mick
parent 5c75907e62
commit 19c16748ce
11 changed files with 469 additions and 377 deletions
@@ -227,7 +227,6 @@ TI2I_sampling_params = DiffusionSamplingParams(
T2V_PROMPT = "A curious raccoon"
TI2V_sampling_params = DiffusionSamplingParams(
output_size="832x1104",
prompt="The man in the picture slowly turns his head, his expression enigmatic and otherworldly. The camera performs a slow, cinematic dolly out, focusing on his face. Moody lighting, neon signs glowing in the background, shallow depth of field.",
image_path="https://is1-ssl.mzstatic.com/image/thumb/Music114/v4/5f/fa/56/5ffa56c2-ea1f-7a17-6bad-192ff9b6476d/825646124206.jpg/600x600bb.jpg",
)
@@ -302,7 +301,6 @@ ONE_GPU_CASES_B: list[DiffusionTestCase] = [
),
DiffusionSamplingParams(
prompt=T2V_PROMPT,
output_size="848x480",
),
),
# LoRA test case for single transformer + merge/unmerge API test
@@ -319,7 +317,6 @@ ONE_GPU_CASES_B: list[DiffusionTestCase] = [
),
DiffusionSamplingParams(
prompt="csetiarcane Nfj1nx with blue hair, a woman walking in a cyberpunk city at night",
output_size="480x320",
num_frames=8,
),
),
@@ -355,7 +352,6 @@ ONE_GPU_CASES_B: list[DiffusionTestCase] = [
),
DiffusionSamplingParams(
prompt=T2V_PROMPT,
output_size="720x480",
),
),
# === Text and Image to Video (TI2V) ===
@@ -384,7 +380,6 @@ ONE_GPU_CASES_B: list[DiffusionTestCase] = [
]
TWO_GPU_CASES_A = [
# TODO: Timeout with Torch2.9. Add back when it can pass CI
DiffusionTestCase(
"wan2_2_i2v_a14b_2gpu",
DiffusionServerArgs(
@@ -408,7 +403,6 @@ TWO_GPU_CASES_A = [
),
DiffusionSamplingParams(
prompt=T2V_PROMPT,
output_size="720x480",
),
),
# LoRA test case for transformer_2 support
@@ -425,7 +419,6 @@ TWO_GPU_CASES_A = [
),
DiffusionSamplingParams(
prompt="Nfj1nx with blue hair, a woman walking in a cyberpunk city at night",
output_size="720x480",
),
),
DiffusionTestCase(
@@ -440,7 +433,7 @@ TWO_GPU_CASES_A = [
),
DiffusionSamplingParams(
prompt=T2V_PROMPT,
output_size="720x480",
output_size="832x480",
),
),
]