Generate: unify LogitsWarper and LogitsProcessor (#32626)

2024-08-16 11:20:41 +01:00
parent 5fd7ca7bc9
commit 70d5df6107
20 changed files with 186 additions and 623 deletions
--- a/tests/models/mamba/test_modeling_mamba.py
+++ b/tests/models/mamba/test_modeling_mamba.py
@@ -433,6 +433,10 @@ class MambaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi
            dict_inputs = self._prepare_for_class(inputs_dict, model_class, return_labels=True)
            check_equivalence(model, tuple_inputs, dict_inputs, {"output_hidden_states": True})

+    @unittest.skip("The `input_embeds` when fed don't produce the same results.")
+    def test_beam_sample_generate(self):
+        pass
+

@require_torch
 class MambaIntegrationTests(unittest.TestCase):