Skip to content

GitLab

Commit 4f75d380 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

make style

parent c2ee3840
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/summarization/bart/evaluate_cnn.py
View file @ 4f75d380
......@@ -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:
......
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