Peter commited on
Commit
131a91a
1 Parent(s): 281a517

remove extra line

Browse files
Files changed (1) hide show
  1. summarize.py +0 -1
summarize.py CHANGED
@@ -46,7 +46,6 @@ def summarize_and_score(ids, mask, model, tokenizer, **kwargs):
46
  input_ids = ids.to("cuda") if torch.cuda.is_available() else ids
47
  attention_mask = mask.to("cuda") if torch.cuda.is_available() else mask
48
 
49
- attention_mask = mask.to("cuda")
50
  global_attention_mask = torch.zeros_like(attention_mask)
51
  # put global attention on <s> token
52
  global_attention_mask[:, 0] = 1
 
46
  input_ids = ids.to("cuda") if torch.cuda.is_available() else ids
47
  attention_mask = mask.to("cuda") if torch.cuda.is_available() else mask
48
 
 
49
  global_attention_mask = torch.zeros_like(attention_mask)
50
  # put global attention on <s> token
51
  global_attention_mask[:, 0] = 1