zhihan1996 commited on
Commit
7e37201
1 Parent(s): 782b462

Update bert_layers.py

Browse files
Files changed (1) hide show
  1. bert_layers.py +2 -1
bert_layers.py CHANGED
@@ -18,7 +18,8 @@ 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.models.bert.modeling_bert import BertPreTrainedModel, PreTrainedModel
 
22
 
23
  from .bert_padding import (index_first_axis,
24
  index_put_first_axis, pad_input,
 
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,
25
  index_put_first_axis, pad_input,