From b5003f5806fb40935bdb24158278c5c67db9a2e9 Mon Sep 17 00:00:00 2001 From: younesbelkada Date: Wed, 25 Jan 2023 12:30:53 +0000 Subject: [PATCH] fix blenderbot doctest - add correct expected value --- src/transformers/models/blenderbot/modeling_blenderbot.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/models/blenderbot/modeling_blenderbot.py b/src/transformers/models/blenderbot/modeling_blenderbot.py index 830fb04bda58..1f2b3a878d83 100755 --- a/src/transformers/models/blenderbot/modeling_blenderbot.py +++ b/src/transformers/models/blenderbot/modeling_blenderbot.py @@ -544,7 +544,7 @@ def dummy_inputs(self): >>> inputs = tokenizer([NEXT_UTTERANCE], return_tensors="pt") >>> next_reply_ids = model.generate(**inputs) >>> print("Bot: ", tokenizer.batch_decode(next_reply_ids, skip_special_tokens=True)[0]) - Bot: That's too bad. Have you tried encouraging them to change their eating habits? + Bot: I see. Well, it's good that they're trying to change their eating habits. ``` """