zhihan1996 commited on
Commit
dd10f74
1 Parent(s): 1ce72a5

Update bert_layers.py

Browse files
Files changed (1) hide show
  1. bert_layers.py +1 -1
bert_layers.py CHANGED
@@ -18,7 +18,7 @@ from torch.nn.modules.utils import consume_prefix_in_state_dict_if_present
18
  from transformers.activations import ACT2FN
19
  from transformers.modeling_outputs import (MaskedLMOutput,
20
  SequenceClassifierOutput)
21
- from transformers.modeling_bert import BertPreTrainedModel
22
  from transformers.modeling_utils import PreTrainedModel
23
 
24
  from .bert_padding import (index_first_axis,
 
18
  from transformers.activations import ACT2FN
19
  from transformers.modeling_outputs import (MaskedLMOutput,
20
  SequenceClassifierOutput)
21
+ from transformers.models.bert.modeling_bert import BertPreTrainedModel
22
  from transformers.modeling_utils import PreTrainedModel
23
 
24
  from .bert_padding import (index_first_axis,