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()
|