🚨🚨 Generate: standardize beam search behavior across frameworks (#21368)

2023-02-03 10:24:02 +00:00
parent ea55bd86b9
commit f21af26279
10 changed files with 122 additions and 118 deletions
--- a/tests/models/gpt2/test_modeling_flax_gpt2.py
+++ b/tests/models/gpt2/test_modeling_flax_gpt2.py
@@ -224,7 +224,7 @@ class FlaxGPT2ModelTest(FlaxModelTesterMixin, FlaxGenerationTesterMixin, unittes
        output_string = tokenizer.batch_decode(output_sequences, skip_special_tokens=True)

        expected_string = [
-            "Hello this is a long string of words. I'm going to try to explain what I mean.",
+            "Hello this is a long string of words. I'm going to start with the first one.\n",
            "Hey, I'm not sure if I'm going to be able to do",
        ]