Fix import error in script to convert faisreq roberta checkpoints

49cba6e5 · Louis MARTIN · a701c9b3 · 49cba6e5
Commit 49cba6e5 authored Oct 14, 2019 by Louis MARTIN
Show whitespace changes
Inline Side-by-side

Showing with 9 additions and 9 deletions

transformers/convert_roberta_original_pytorch_checkpoint_to_pytorch.py ...convert_roberta_original_pytorch_checkpoint_to_pytorch.py +9 -9

No files found.
--- a/transformers/convert_roberta_original_pytorch_checkpoint_to_pytorch.py
+++ b/transformers/convert_roberta_original_pytorch_checkpoint_to_pytorch.py
@@ -23,12 +23,12 @@ import torch

 from fairseq.models.roberta import RobertaModel as FairseqRobertaModel
 from fairseq.modules import TransformerSentenceEncoderLayer
-from transformers import (BertConfig, BertEncoder,
+from transformers.modeling_bert import (BertConfig, BertEncoder,
                                        BertIntermediate, BertLayer,
                                        BertModel, BertOutput,
                                        BertSelfAttention,
                                        BertSelfOutput)
-from transformers import (RobertaEmbeddings,
+from transformers.modeling_roberta import (RobertaEmbeddings,
                                           RobertaForMaskedLM,
                                           RobertaForSequenceClassification,
                                           RobertaModel)