Update configuration_bert.py
Browse files- 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 |
-
|
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()
|