Spaces:
Sleeping
Sleeping
Peter
commited on
Commit
•
131a91a
1
Parent(s):
281a517
remove extra line
Browse files- 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
|