riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 19:21:31 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

[Doctest] Fix Blenderbot doctest (#21297)

Browse Source 
fix blenderbot doctest

- add correct expected value
This commit is contained in:
Younes Belkada 2023-01-25 17:28:29 +01:00 committed by GitHub
parent cc714d74c4
commit 015443f42b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/blenderbot/modeling_blenderbot.py
View File

@ -544,7 +544,7 @@ BLENDERBOT_GENERATION_EXAMPLE = r"""
>>> 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.
```
"""