[fix] Disable task specific evaluation

2026-09-12 12:03:06 +08:00 · 2023-01-14 06:47:21 +00:00
parent 1546111094
commit 6f6c590e57
2 changed files with 30 additions and 28 deletions
@@ -8,6 +8,7 @@ SUMMARIZATION_SPECIAL_TOKENS = {"Text": "", "Summary": ["TL;DR:", "Summarize thi
 SUMMARY_SPECIAL_PROMPT = {
    "multi_news": ["Summarize in bullet points", "Generate summary in list of points"],
    "xsum": ["Give me summary in one sentence", "Short TLDR", "Give me a concise summary"],
+    "samsum": ["TLDR;", "Summarize this dialogue", "Summarize dialogue"],
 }

 summarization_config_mapping = {