Unverified Commit f0fc0aea authored by Sam Shleifer's avatar Sam Shleifer Committed by GitHub
Browse files

pegasus.rst: fix expected output (#7017)

parent 120176ea
...@@ -60,7 +60,7 @@ Usage Example ...@@ -60,7 +60,7 @@ Usage Example
batch = tokenizer.prepare_seq2seq_batch(src_text, truncation=True, padding='longest').to(torch_device) batch = tokenizer.prepare_seq2seq_batch(src_text, truncation=True, padding='longest').to(torch_device)
translated = model.generate(**batch) translated = model.generate(**batch)
tgt_text = tokenizer.batch_decode(translated, skip_special_tokens=True) 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 PegasusForConditionalGeneration
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
......
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