Skip to content

Commit f2cd562

Browse files
committed
fix
1 parent 8390b01 commit f2cd562

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/models/glm4_moe/test_modeling_glm4_moe.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@ class Glm4MoeModelTest(CausalLMModelTest, unittest.TestCase):
6161
model_tester_class = Glm4MoeModelTester
6262
# used in `test_torch_compile_for_training`. Skip as "Dynamic control flow in MoE"
6363
_torch_compile_train_cls = None
64+
model_split_percents = [0.5, 0.85, 0.9] # it tries to offload everything with the default value
6465

6566

6667
@require_torch_accelerator

0 commit comments

Comments
 (0)