Generation: fix test (#34369)

* fix test * fix copies
2024-10-29 07:57:10 +01:00
parent fe76b60370
commit 808d6c50f8
4 changed files with 28 additions and 44 deletions
--- a/tests/models/mamba2/test_modeling_mamba2.py
+++ b/tests/models/mamba2/test_modeling_mamba2.py
@@ -204,7 +204,8 @@ class Mamba2ModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMix
        pass

    @unittest.skip(reason="To fix, Mamba 2 cache slicing test case is an edge case")
-    def test_generate_from_inputs_embeds_decoder_only(self):
+    @parameterized.expand([(1,), (2,)])
+    def test_generate_from_inputs_embeds_decoder_only(self, num_beams):
        pass

    @unittest.skip(reason="To fix, Mamba 2 cache slicing test case is an edge case")