Add summarization name mapping for MultiNews (#18117)

* Add summarization name mapping for MultiNews * Add summarization name mapping for MultiNews
2022-07-13 05:19:20 -07:00
parent 195133363e
commit fde22c75a1
2 changed files with 2 additions and 0 deletions
--- a/examples/pytorch/summarization/run_summarization.py
+++ b/examples/pytorch/summarization/run_summarization.py
@@ -286,6 +286,7 @@ summarization_name_mapping = {
    "xglue": ("news_body", "news_title"),
    "xsum": ("document", "summary"),
    "wiki_summary": ("article", "highlights"),
+    "multi_news": ("document", "summary"),
 }