From 4f75d380a46f343b0c2244eff60484ed2a85ec16 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Fri, 13 Mar 2020 16:35:52 +0100 Subject: [PATCH] make style --- examples/summarization/bart/evaluate_cnn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/summarization/bart/evaluate_cnn.py b/examples/summarization/bart/evaluate_cnn.py index 93590614eb..fded7e51f0 100644 --- a/examples/summarization/bart/evaluate_cnn.py +++ b/examples/summarization/bart/evaluate_cnn.py @@ -36,7 +36,7 @@ def generate_summaries(lns, out_file, batch_size=8, device=DEFAULT_DEVICE): no_repeat_ngram_size=3, early_stopping=True, do_sample=False, - decoder_start_token_id=model.config.eos_token_ids[0] + decoder_start_token_id=model.config.eos_token_ids[0], ) dec = [tokenizer.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=False) for g in summaries] for hypothesis in dec: