From f0fc0aea6bd885ee90837cd96c3935a5d44e060a Mon Sep 17 00:00:00 2001 From: Sam Shleifer Date: Tue, 8 Sep 2020 13:29:16 -0400 Subject: [PATCH] pegasus.rst: fix expected output (#7017) --- docs/source/model_doc/pegasus.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/model_doc/pegasus.rst b/docs/source/model_doc/pegasus.rst index 852c03380c..ca48f18105 100644 --- a/docs/source/model_doc/pegasus.rst +++ b/docs/source/model_doc/pegasus.rst @@ -60,7 +60,7 @@ Usage Example batch = tokenizer.prepare_seq2seq_batch(src_text, truncation=True, padding='longest').to(torch_device) translated = model.generate(**batch) tgt_text = tokenizer.batch_decode(translated, skip_special_tokens=True) - assert tgt_text[0] == "California's largest electricity provider has turned off power to tens of thousands of customers." + assert tgt_text[0] == "California's largest electricity provider has turned off power to hundreds of thousands of customers." PegasusForConditionalGeneration ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~