zhihan1996 commited on
Commit
55b04c5
1 Parent(s): 0109f62

Update configuration_bert.py

Browse files
Files changed (1) hide show
  1. configuration_bert.py +2 -2
configuration_bert.py CHANGED
@@ -1,10 +1,10 @@
1
  # Copyright 2022 MosaicML Examples authors
2
  # SPDX-License-Identifier: Apache-2.0
3
 
4
- from transformers import BertConfig as TransformersBertConfig
5
 
6
 
7
- class BertConfig(TransformersBertConfig):
8
 
9
  def __init__(
10
  self,
 
1
  # Copyright 2022 MosaicML Examples authors
2
  # SPDX-License-Identifier: Apache-2.0
3
 
4
+ from transformers.configuration_utils import PretrainedConfig
5
 
6
 
7
+ class BertConfig(PretrainedConfig):
8
 
9
  def __init__(
10
  self,