Revert D71058400

Summary:
This diff reverts D71058400
broke spna/bender_predictor:bender_predictor_lib-type-checking - main T218443607

Differential Revision: D71478648

fbshipit-source-id: 6a2017f32fd219a19ea6719fd083dff79034a1a6

Author: generatedunixname89002005232357

mmf/modules/hf_layers.py

@@ -8,8 +8,7 @@
 from torch import nn, Tensor
 
 try:
-    from transformers.modeling_utils import PreTrainedModel
-    from transformers.models.bert.modeling_bert import (
+    from transformers3.modeling_bert import (
         BertAttention,
         BertEmbeddings,
         BertEncoder,
@@ -19,17 +18,17 @@
         BertSelfAttention,
         BertSelfOutput,
     )
-    from transformers.models.roberta.modeling_roberta import (
+    from transformers3.modeling_roberta import (
         RobertaAttention,
         RobertaEmbeddings,
         RobertaEncoder,
         RobertaLayer,
         RobertaModel,
         RobertaSelfAttention,
     )
+    from transformers3.modeling_utils import PreTrainedModel
 except ImportError:
-    from transformers.modeling_utils import PreTrainedModel
-    from transformers.models.bert.modeling_bert import (
+    from transformers.modeling_bert import (
         BertAttention,
         BertEmbeddings,
         BertEncoder,
@@ -39,14 +38,15 @@
         BertSelfAttention,
         BertSelfOutput,
     )
-    from transformers.models.roberta.modeling_roberta import (
+    from transformers.modeling_roberta import (
         RobertaAttention,
         RobertaEmbeddings,
         RobertaEncoder,
         RobertaLayer,
         RobertaModel,
         RobertaSelfAttention,
     )
+    from transformers.modeling_utils import PreTrainedModel
 
 
 patch_functions = [