frugal-ai-submission-template

Sleeping

Tonic commited on Feb 10

Commit

ebc55a1

unverified ·

1 Parent(s): a8d5c7d

add modernbert config

Files changed (1) hide show

tasks/text.py CHANGED Viewed

@@ -3,7 +3,7 @@ from datetime import datetime
 from datasets import load_dataset
 from sklearn.metrics import accuracy_score
 import torch
-from transformers import AutoTokenizer, AutoModelForSequenceClassification, AutoConfig, PretrainedConfig
 from torch.utils.data import DataLoader
 from transformers import DataCollatorWithPadding

 from datasets import load_dataset
 from sklearn.metrics import accuracy_score
 import torch
+from transformers import AutoTokenizer, AutoModelForSequenceClassification, ModernBertConfig
 from torch.utils.data import DataLoader
 from transformers import DataCollatorWithPadding