Clean imports in flaubert_model.py.

ThilinaRajapakse · Feb 10, 2020 · 2cfbe5f · 2cfbe5f
1 parent 0ae8834
commit 2cfbe5f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/simpletransformers/classification/transformer_models/flaubert_model.py b/simpletransformers/classification/transformer_models/flaubert_model.py
@@ -1,4 +1,4 @@
-from transformers.modeling_xlm import SequenceSummary, XLMModel, XLMPreTrainedModel
+from transformers.modeling_utils import SequenceSummary
 from transformers.modeling_flaubert import FlaubertModel
 import torch
 import torch.nn as nn