Update config.json

config.json

@@ -1,15 +1,20 @@
 {
+  "amp": 1,
   "architectures": [
     "FlaubertWithLMHeadModel"
   ],
-  "amp": 1,
+  "asm": false,
   "attention_dropout": 0.1,
   "bos_index": 0,
+  "bos_token_id": 0,
   "bptt": 512,
+  "causal": false,
   "clip_grad_norm": 5,
   "dropout": 0.1,
   "emb_dim": 768,
+  "embed_init_std": 0.02209708691207961,
   "encoder_only": true,
+  "end_n_top": 5,
   "eos_index": 1,
   "fp16": true,
   "gelu_activation": true,
@@ -17,16 +22,23 @@
   "id2lang": {
     "0": "fr"
   },
+  "init_std": 0.02,
+  "is_encoder": true,
   "lang2id": {
     "fr": 0
   },
+  "lang_id": 0,
   "langs": [
     "fr"
   ],
+  "layer_norm_eps": 1e-12,
+  "layerdrop": 0.0,
   "lg_sampling_factor": -1,
   "lgs": "fr",
   "mask_index": 5,
+  "mask_token_id": 0,
   "max_batch_size": 0,
+  "max_position_embeddings": 512,
   "max_vocab": -1,
   "mlm_steps": [
     [
@@ -34,16 +46,26 @@
       null
     ]
   ],
+  "model_type": "flaubert",
   "n_heads": 12,
   "n_langs": 1,
   "n_layers": 12,
-  "n_words": 68729,
   "pad_index": 2,
+  "pad_token_id": 2,
+  "pre_norm": false,
   "sample_alpha": 0,
   "share_inout_emb": true,
+  "sinusoidal_embeddings": false,
+  "start_n_top": 5,
+  "summary_activation": null,
+  "summary_first_dropout": 0.1,
+  "summary_proj_to_labels": true,
+  "summary_type": "first",
+  "summary_use_proj": true,
   "tokens_per_batch": -1,
   "unk_index": 3,
   "use_lang_emb": true,
+  "vocab_size": 68729,
   "word_blank": 0,
   "word_dropout": 0,
   "word_keep": 0.1,