ai-forever commited on
Commit
af37edf
1 Parent(s): 5d2cd7b

Update config.json

Browse files

fix warnings with task type

Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "architectures": [
3
- "ElectraModel"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
6
  "embedding_size": 128,
 
1
  {
2
  "architectures": [
3
+ "ElectraForPreTraining"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
6
  "embedding_size": 128,