[CI][Bugfix] Fix Minimax-H3 FP8 Accuracy Test (#5829)
Signed-off-by: mglyn <1203789601@qq.com>
This commit is contained in:
1 parent
9235b0ae65
commit
240e6655be
1 file changed
+1
@@ -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)
|
||||
|
||||
Reference in new issue
Block a user