Generate: TF sample doctest result update (#20208)

7b55bb45 · Joao Gante · GitHub · d24e84d9 · 7b55bb45
Unverified Commit 7b55bb45 authored Nov 14, 2022 by Joao Gante Committed by GitHub Nov 14, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/generation/tf_utils.py src/transformers/generation/tf_utils.py +1 -1

No files found.
--- a/src/transformers/generation/tf_utils.py
+++ b/src/transformers/generation/tf_utils.py
@@ -2671,7 +2671,7 @@ class TFGenerationMixin:
        >>> outputs = model.sample(input_ids, logits_processor=logits_processor, logits_warper=logits_warper)

        >>> tokenizer.batch_decode(outputs, skip_special_tokens=True)
-        ['Today is a beautiful day, and it\'s all about the future."\n\nThe announcement comes three']
+        ['Today is a beautiful day, and I love my country. But when I look at Donald Trump,']
        ```"""

        # 1. init greedy_search values