allenai
/

longformer-scico

Text Classification

longformer-scico

Model card Files Files and versions Community

cattana commited on Sep 17, 2021

Commit

dc2071b

•

1 Parent(s): 8e4aa42

fix

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -27,7 +27,7 @@ tokenizer = AutoTokenizer.from_pretrained('allenai/longformer-scico')
 model = AutoModelForSequenceClassification.from_pretrained('allenai/longformer-scico')
 start_token = tokenizer.convert_tokens_to_ids("<m>")
-end_token = tokenizer.convert_tokens_to_aids("</m>")
 def get_global_attention(input_ids):
     global_attention_mask = torch.zeros(input_ids.shape)

 model = AutoModelForSequenceClassification.from_pretrained('allenai/longformer-scico')
 start_token = tokenizer.convert_tokens_to_ids("<m>")
+end_token = tokenizer.convert_tokens_to_ids("</m>")
 def get_global_attention(input_ids):
     global_attention_mask = torch.zeros(input_ids.shape)