Blaise-g commited on
Commit
aadb57a
Β·
1 Parent(s): ece3bd3

Update summarize.py

Browse files
Files changed (1) hide show
  1. 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,