Generate: group_beam_search requires diversity_penalty>0.0 (#24456)

* add exception * update docs
2023-06-27 10:46:39 +01:00
parent 43479ef98f
commit 5f3efdf762
3 changed files with 11 additions and 4 deletions
--- a/tests/generation/test_utils.py
+++ b/tests/generation/test_utils.py
@@ -2366,6 +2366,7 @@ class GenerationIntegrationTests(unittest.TestCase, GenerationIntegrationTestsMi
            num_beams=2,
            num_beam_groups=2,
            num_return_sequences=2,
+            diversity_penalty=1.0,
            eos_token_id=None,
            return_dict_in_generate=True,
            output_scores=True,