Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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
|