[diffusion] fix: fix the frame interpolation testcase in CI regarding number of frames (#19659)

Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
Junhao Liu
2026-03-05 11:21:53 +08:00
committed by GitHub
co-authored by Mick
parent fc53307ce9
commit 9c11a7ae40
2 changed files with 59 additions and 61 deletions
@@ -1988,69 +1988,68 @@
}, },
"wan2_1_t2v_1.3b_frame_interp_2x": { "wan2_1_t2v_1.3b_frame_interp_2x": {
"stages_ms": { "stages_ms": {
"InputValidationStage": 0.03, "TextEncodingStage": 1104.4,
"TextEncodingStage": 1155.78, "TimestepPreparationStage": 2.19,
"LatentPreparationStage": 0.12, "LatentPreparationStage": 0.15,
"TimestepPreparationStage": 2.17, "DenoisingStage": 8502.22,
"DenoisingStage": 4977.09, "DecodingStage": 498.36,
"DecodingStage": 98.46, "InputValidationStage": 0.07
"per_frame_generation": 1271.97
}, },
"denoise_step_ms": { "denoise_step_ms": {
"0": 85.33, "0": 91.83,
"1": 76.41, "1": 174.57,
"2": 100.5, "2": 170.48,
"3": 99.28, "3": 169.33,
"4": 100.14, "4": 169.24,
"5": 100.33, "5": 177.43,
"6": 100.17, "6": 173.73,
"7": 100.35, "7": 171.67,
"8": 100.14, "8": 170.98,
"9": 100.32, "9": 168.61,
"10": 100.18, "10": 169.96,
"11": 100.37, "11": 174.75,
"12": 100.17, "12": 172.33,
"13": 100.34, "13": 170.62,
"14": 100.21, "14": 169.84,
"15": 100.35, "15": 168.86,
"16": 100.15, "16": 171.32,
"17": 100.32, "17": 174.7,
"18": 100.23, "18": 172.31,
"19": 100.35, "19": 171.71,
"20": 100.19, "20": 170.98,
"21": 100.57, "21": 169.83,
"22": 100.27, "22": 170.54,
"23": 100.24, "23": 173.08,
"24": 100.27, "24": 172.11,
"25": 100.27, "25": 171.49,
"26": 100.27, "26": 171.0,
"27": 100.29, "27": 170.9,
"28": 100.25, "28": 171.78,
"29": 100.25, "29": 173.44,
"30": 100.3, "30": 171.14,
"31": 100.3, "31": 170.72,
"32": 100.23, "32": 170.64,
"33": 100.25, "33": 170.58,
"34": 100.26, "34": 172.51,
"35": 100.27, "35": 171.74,
"36": 100.24, "36": 171.57,
"37": 100.26, "37": 170.73,
"38": 100.26, "38": 171.49,
"39": 100.29, "39": 170.98,
"40": 100.31, "40": 172.63,
"41": 100.31, "41": 171.88,
"42": 100.31, "42": 171.71,
"43": 100.26, "43": 170.94,
"44": 100.26, "44": 170.31,
"45": 100.27, "45": 171.25,
"46": 100.3, "46": 171.43,
"47": 100.27, "47": 171.55,
"48": 100.23, "48": 172.08,
"49": 99.65 "49": 169.92
}, },
"expected_e2e_ms": 6359.87, "expected_e2e_ms": 10464.97,
"expected_avg_denoise_ms": 99.47, "expected_avg_denoise_ms": 169.92,
"expected_median_denoise_ms": 100.27 "expected_median_denoise_ms": 171.37
}, },
"flux_2_klein_ti2i_2_gpus": { "flux_2_klein_ti2i_2_gpus": {
"stages_ms": { "stages_ms": {
@@ -516,7 +516,6 @@ ONE_GPU_CASES_B: list[DiffusionTestCase] = [
), ),
DiffusionSamplingParams( DiffusionSamplingParams(
prompt=T2V_PROMPT, prompt=T2V_PROMPT,
num_frames=5,
enable_frame_interpolation=True, enable_frame_interpolation=True,
frame_interpolation_exp=1, frame_interpolation_exp=1,
), ),