SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Skip failing JetMOE generation tests (#31266)

Browse Source 
Skip failing tests for now
This commit is contained in:
amyeroberts
2024-06-05 19:06:46 +01:00
committed by GitHub
parent bd5091df8d
commit 940fde8daf
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/models/jetmoe/test_modeling_jetmoe.py
View File

@@ -472,6 +472,14 @@ class JetMoeModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMix
def test_flash_attn_2_inference_equivalence_right_padding(self): def test_flash_attn_2_inference_equivalence_right_padding(self):
self.skipTest("JetMoe flash attention does not support right padding") 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 @require_torch
class JetMoeIntegrationTest(unittest.TestCase): class JetMoeIntegrationTest(unittest.TestCase):