[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:
@@ -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",
|
||||
),
|
||||
),
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user