From fde22c75a157a9f8c2da620a3b3bf9a23b5b3a0a Mon Sep 17 00:00:00 2001 From: John Giorgi Date: Wed, 13 Jul 2022 05:19:20 -0700 Subject: [PATCH] Add summarization name mapping for MultiNews (#18117) * Add summarization name mapping for MultiNews * Add summarization name mapping for MultiNews --- examples/pytorch/summarization/run_summarization.py | 1 + examples/tensorflow/summarization/run_summarization.py | 1 + 2 files changed, 2 insertions(+) diff --git a/examples/pytorch/summarization/run_summarization.py b/examples/pytorch/summarization/run_summarization.py index 70a7f81b5a..8d5007a503 100755 --- 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"), } diff --git a/examples/tensorflow/summarization/run_summarization.py b/examples/tensorflow/summarization/run_summarization.py index 4da3438289..fdeb3f6249 100644 --- a/examples/tensorflow/summarization/run_summarization.py +++ b/examples/tensorflow/summarization/run_summarization.py @@ -266,6 +266,7 @@ summarization_name_mapping = { "xglue": ("news_body", "news_title"), "xsum": ("document", "summary"), "wiki_summary": ("article", "highlights"), + "multi_news": ("document", "summary"), } # endregion