fix
Browse files
README.md
CHANGED
@@ -27,7 +27,7 @@ tokenizer = AutoTokenizer.from_pretrained('allenai/longformer-scico')
|
|
27 |
model = AutoModelForSequenceClassification.from_pretrained('allenai/longformer-scico')
|
28 |
|
29 |
start_token = tokenizer.convert_tokens_to_ids("<m>")
|
30 |
-
end_token = tokenizer.
|
31 |
|
32 |
def get_global_attention(input_ids):
|
33 |
global_attention_mask = torch.zeros(input_ids.shape)
|
|
|
27 |
model = AutoModelForSequenceClassification.from_pretrained('allenai/longformer-scico')
|
28 |
|
29 |
start_token = tokenizer.convert_tokens_to_ids("<m>")
|
30 |
+
end_token = tokenizer.convert_tokens_to_ids("</m>")
|
31 |
|
32 |
def get_global_attention(input_ids):
|
33 |
global_attention_mask = torch.zeros(input_ids.shape)
|