monologg commited on
Commit
e6196a5
1 Parent(s): 6b3a739

docs: inference api template

Browse files
Files changed (2) hide show
  1. README.md +3 -0
  2. config.json +1 -1
README.md CHANGED
@@ -3,6 +3,9 @@ language: ko
3
  tags:
4
  - korean
5
  - klue
 
 
 
6
  ---
7
 
8
  # KLUE BERT base
3
  tags:
4
  - korean
5
  - klue
6
+ mask_token: "[MASK]"
7
+ widget:
8
+ - text: 대한민국의 수도는 [MASK] 입니다.
9
  ---
10
 
11
  # KLUE BERT base
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "architectures": ["BertForPretraining"],
3
  "attention_probs_dropout_prob": 0.1,
4
  "hidden_act": "gelu",
5
  "hidden_dropout_prob": 0.1,
1
  {
2
+ "architectures": ["BertForMaskedLM"],
3
  "attention_probs_dropout_prob": 0.1,
4
  "hidden_act": "gelu",
5
  "hidden_dropout_prob": 0.1,