Modify architecture in config.json from BertForPreTraining to BertForMaskedLM to allow inference API to estimate a pipeline type

CHANGELOG.md

@@ -5,3 +5,8 @@
 ### Added
 
 - models and model card
+
+### Modified
+
+- architecture in config.json from BertForPreTraining to BertForMaskedLM to allow inference API to estimate a pipeline type
+

README.md

@@ -58,7 +58,7 @@ Training code is available in [a GitHub repository](https://github.com/colorfuls
 First, install dependecies.
 
 ```sh
-$ pip install torch==1.8.0 transformers==4.8.2 sentencepiece==0.1.95 tensorflow==2.5.0
+$ pip install torch==1.8.0 transformers==4.8.2 sentencepiece==0.1.95
 ```
 
 Then use `transformers.pipeline` to try mask fill task.

config.json

@@ -1,7 +1,7 @@
 {
   "_name_or_path": "release/bert-base-ja",
   "architectures": [
-    "BertForPreTraining"
+    "BertForMaskedLM"
   ],
   "attention_probs_dropout_prob": 0.1,
   "bos_token_id": 2,