Update summarize.py

summarize.py

@@ -53,6 +53,7 @@ def summarize(ids, mask, model, tokenizer, **kwargs):
         input_ids,
         attention_mask=attention_mask,
         #global_attention_mask=global_attention_mask,
+        return_dict_in_generate=True,
         **kwargs,
     )
     summary = tokenizer.batch_decode(