[CI][bugfix] Fix MiniMax H3 FP8 quality test remote-code loading (#7617)
This commit is contained in:
1 parent
d81777f297
commit
72cc012cfd
1 file changed
+2
@@ -189,6 +189,8 @@ def test_minimax_h3_quantization_quality(config: QualityTestConfig):
|
||||
# The fused BF16 baseline only fits on H100-80GB with encoder TP.
|
||||
"stage_0_tensor_parallel_size": 2,
|
||||
"vae_use_tiling": True,
|
||||
# MiniMax H3's VAE is implemented by code shipped with the checkpoint.
|
||||
"trust_remote_code": True,
|
||||
}
|
||||
|
||||
omni_bl = Omni(**common_kwargs)
|
||||
|
||||
Reference in new issue
Block a user