Skip to content

GitLab

Unverified Commit 7b55bb45 authored by Joao Gante's avatar Joao Gante Committed by GitHub
Browse files

Generate: TF sample doctest result update (#20208)

parent d24e84d9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/generation/tf_utils.py
View file @ 7b55bb45
...@@ -2671,7 +2671,7 @@ class TFGenerationMixin: ...@@ -2671,7 +2671,7 @@ class TFGenerationMixin:
>>> outputs = model.sample(input_ids, logits_processor=logits_processor, logits_warper=logits_warper) >>> outputs = model.sample(input_ids, logits_processor=logits_processor, logits_warper=logits_warper)
>>> tokenizer.batch_decode(outputs, skip_special_tokens=True) >>> 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 # 1. init greedy_search values
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment