Fix low memory beam search (#34746)

* fix * higher max positions in tests
2024-11-20 07:46:35 +01:00
parent 145fbd46cb
commit 9470d65324
3 changed files with 10 additions and 5 deletions
--- a/tests/generation/test_utils.py
+++ b/tests/generation/test_utils.py
@@ -1046,7 +1046,6 @@ class GenerationTesterMixin:
            self.assertListEqual(low_output.tolist(), high_output.tolist())

    @pytest.mark.generate
-    @unittest.skip("Started to break with https://github.com/huggingface/transformers/pull/33703")
    def test_beam_search_low_memory(self):
        # Check that choosing 'low_memory' does not change the model output
        for model_class in self.all_generative_model_classes: