Update summarize.py
Browse files- summarize.py +1 -1
summarize.py
CHANGED
@@ -52,7 +52,7 @@ def summarize_and_score(ids, mask, model, tokenizer, **kwargs):
|
|
52 |
summary_pred_ids = model.generate(
|
53 |
input_ids,
|
54 |
attention_mask=attention_mask,
|
55 |
-
global_attention_mask=global_attention_mask,
|
56 |
output_scores=True,
|
57 |
return_dict_in_generate=True,
|
58 |
**kwargs,
|
|
|
52 |
summary_pred_ids = model.generate(
|
53 |
input_ids,
|
54 |
attention_mask=attention_mask,
|
55 |
+
#global_attention_mask=global_attention_mask,
|
56 |
output_scores=True,
|
57 |
return_dict_in_generate=True,
|
58 |
**kwargs,
|