Update tasks/text.py

tasks/text.py

@@ -12,7 +12,7 @@ from sentence_transformers import SentenceTransformer
 import torch
 import torch.nn as nn
 from torch.utils.data import TensorDataset, DataLoader, RandomSampler, SequentialSampler
-from transformers import BertForPreTraining, BertModel, AutoTokenizer, BertForSequenceClassification, RobertaForSequenceClassification
+from transformers import BertForPreTraining, BertModel, AutoTokenizer, AutoModel, ModernBertForSequenceClassification, BertForSequenceClassification, RobertaForSequenceClassification
 
 
 from .utils.evaluation import TextEvaluationRequest