correct config.json

config.json

@@ -1,25 +1,19 @@
 {
   "_name_or_path": "juenp/FrenchTextCategirizer",
-  "amp": 1,
   "architectures": [
     "FlaubertForSequenceClassification"
   ],
-  "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,
-  "group_by_size": true,
+  "eos_token_id": 1,
+  "pad_token_id": 2,
+  "unk_token_id": 3,
+  "hidden_size": 256,  // Modifié pour correspondre à hidden_dim: 256
+  "layer_norm_eps": 1e-12,
+  "n_heads": 12,
+  "n_layers": 12,
+  "vocab_size": 68729,
+  "model_type": "flaubert",
   "id2label": {
     "0": "CULTURE",
     "1": "DEBATS_ET_OPINIONS",
@@ -37,18 +31,12 @@
     "13": "SPORT",
     "14": "INDEFINI"
   },
-  "id2lang": {
-    "0": "fr"
-  },
-  "init_std": 0.02,
-  "is_encoder": true,
   "label2id": {
     "CULTURE": 0,
     "DEBATS_ET_OPINIONS": 1,
     "ECONOMIE": 2,
     "EDUCATION": 3,
     "FAIT_DIVERS": 4,
-    "INDEFINI": 14,
     "INTERNATIONAL": 5,
     "LIFESTYLE": 6,
     "NUMERIQUE": 7,
@@ -57,58 +45,11 @@
     "SANTE": 10,
     "SCIENCE_ET_ENVIRONNEMENT": 11,
     "SOCIETE": 12,
-    "SPORT": 13
-  },
-  "lang2id": {
-    "fr": 0
+    "SPORT": 13,
+    "INDEFINI": 14
   },
-  "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": [
-    [
-      "fr",
-      null
-    ]
-  ],
-  "model_type": "flaubert",
-  "n_heads": 12,
-  "n_langs": 1,
-  "n_layers": 12,
-  "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,
-  "torch_dtype": "float32",
+  "num_labels": 15,
   "transformers_version": "4.28.1",
-  "unk_index": 3,
-  "use_lang_emb": true,
-  "vocab_size": 68729,
-  "word_blank": 0,
-  "word_dropout": 0,
-  "word_keep": 0.1,
-  "word_mask": 0.8,
-  "word_mask_keep_rand": "0.8,0.1,0.1",
-  "word_pred": 0.15,
-  "word_rand": 0.1,
-  "word_shuffle": 0
+  "torch_dtype": "float32"
 }
+