Fix MoE gradient test (#38438)

2025-05-28 16:44:20 +01:00
parent 0ed6f7e6b4
commit f844733568
3 changed files with 3 additions and 0 deletions
--- a/tests/models/phimoe/test_modeling_phimoe.py
+++ b/tests/models/phimoe/test_modeling_phimoe.py
@@ -102,6 +102,7 @@ class PhimoeModelTest(CausalLMModelTest, unittest.TestCase):

    test_headmasking = False
    test_pruning = False
+    test_all_params_have_gradient = False
    model_tester_class = PhimoeModelTester
    pipeline_model_mapping = (
        {