robinzixuan commited on
Commit
3abefd7
1 Parent(s): 49ce65d

Update configuration_bert.py

Browse files
Files changed (1) hide show
  1. configuration_bert.py +3 -1
configuration_bert.py CHANGED
@@ -11,6 +11,7 @@
11
  # Unless required by applicable law or agreed to in writing, software
12
  # distributed under the License is distributed on an "AS IS" BASIS,
13
  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 
14
  # See the License for the specific language governing permissions and
15
  # limitations under the License.
16
  """BERT model configuration"""
@@ -24,7 +25,7 @@ from transformers.utils import logging
24
 
25
 
26
  logger = logging.get_logger(__name__)
27
- BertConfig.register_for_auto_class()
28
 
29
 
30
  class BertConfig(PretrainedConfig):
@@ -150,3 +151,4 @@ class BertOnnxConfig(OnnxConfig):
150
  ("token_type_ids", dynamic_axis),
151
  ]
152
  )
 
 
11
  # Unless required by applicable law or agreed to in writing, software
12
  # distributed under the License is distributed on an "AS IS" BASIS,
13
  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14
+
15
  # See the License for the specific language governing permissions and
16
  # limitations under the License.
17
  """BERT model configuration"""
 
25
 
26
 
27
  logger = logging.get_logger(__name__)
28
+
29
 
30
 
31
  class BertConfig(PretrainedConfig):
 
151
  ("token_type_ids", dynamic_axis),
152
  ]
153
  )
154
+ BertConfig.register_for_auto_class()