Skip to content

Commit e5da0ae

Browse files
tarekziadeclaude
andauthored
[minimax_m3_vl] Gate the integration tests on total accelerator memory (#50025)
MiniMaxAI/MiniMax-M3-MXFP8 is 413 GiB of weights. On smaller runners `device_map="auto"` offloads the MoE weights to disk, which fails since they need re-saving. Skip unless 480 GiB of accelerator memory is visible. Fixes #50023 Co-authored-by: Claude Opus 5.5 <noreply@anthropic.com>
1 parent 50a952d commit e5da0ae

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

‎tests/models/minimax_m3_vl/test_modeling_minimax_m3_vl.py‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
)
3333
from transformers.testing_utils import (
3434
require_torch,
35+
require_torch_accelerator_memory,
3536
slow,
3637
torch_device,
3738
)
@@ -380,6 +381,7 @@ def test_three_axes_frequency_ladder(self, head_dim, axis_dim):
380381

381382
@slow
382383
@require_torch
384+
@require_torch_accelerator_memory(memory=480) # 413 GiB of MXFP8 weights, plus activations
383385
class MiniMaxM3VLIntegrationTest(unittest.TestCase):
384386
model_id = "MiniMaxAI/MiniMax-M3-MXFP8"
385387

0 commit comments

Comments
 (0)