remove unused import

IndoNLP · Dec 3, 2022 · 0882bd3 · 0882bd3
1 parent c1f675a
commit 0882bd3
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/modules/word_classification.py b/modules/word_classification.py
@@ -7,8 +7,7 @@
 from torch.nn import CrossEntropyLoss, MSELoss
 
 from transformers.modeling_utils import PreTrainedModel, prune_linear_layer
-from transformers.modeling_xlm import XLMPreTrainedModel
-from transformers import AlbertPreTrainedModel, BertPreTrainedModel, AlbertModel, BertModel, BertConfig, XLMModel, XLMConfig, XLMRobertaModel, XLMRobertaConfig
+from transformers import AlbertPreTrainedModel, BertPreTrainedModel, XLMPreTrainedModel, AlbertModel, BertModel, BertConfig, XLMModel, XLMConfig, XLMRobertaModel, XLMRobertaConfig
 from transformers import AutoTokenizer, AutoConfig
 
 XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_MAP = {