Fix jetmoe model (#31279)

* Fix jetmoe model * Remove skip-tests
2024-06-07 11:51:41 +02:00
parent f868cf731a
commit 8bcf9c8dd4
2 changed files with 9 additions and 21 deletions
--- a/tests/models/jetmoe/test_modeling_jetmoe.py
+++ b/tests/models/jetmoe/test_modeling_jetmoe.py
@@ -472,14 +472,6 @@ class JetMoeModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMix
    def test_flash_attn_2_inference_equivalence_right_padding(self):
        self.skipTest("JetMoe flash attention does not support right padding")

-    @unittest.skip("TODO: @ArthurZucker - Breaks after #30536 ")
-    def test_beam_sample_generate(self):
-        pass
-
-    @unittest.skip("TODO: @ArthurZucker - Breaks after #30536 ")
-    def test_generate_from_inputs_embeds_decoder_only(self):
-        pass
-

@require_torch
 class JetMoeIntegrationTest(unittest.TestCase):