diff --git a/tests/diffusion/models/minimax_h3/test_minimax_h3_quantization_quality.py b/tests/diffusion/models/minimax_h3/test_minimax_h3_quantization_quality.py index d93b1c9f8..80d99895f 100644 --- a/tests/diffusion/models/minimax_h3/test_minimax_h3_quantization_quality.py +++ b/tests/diffusion/models/minimax_h3/test_minimax_h3_quantization_quality.py @@ -135,6 +135,7 @@ def _generate_joint_output(omni, config: QualityTestConfig): num_frames=config.num_frames, guidance_scale=config.guidance_scale, sigmas=config.sigmas, + extra_args={"aspect_ratio": "16:9"}, ), ) main_process_peak_mem = torch.accelerator.max_memory_allocated() / (1024**3)