Update config

config.json

@@ -1,15 +1,17 @@
 {
-  "_name_or_path": "/lustre/fswork/projects/rech/gkb/commun/models/pretrained/whisper/whisper-large-v3_dec2_init_cs",
+  "_name_or_path": "",
   "activation_dropout": 0.0,
   "activation_function": "gelu",
-  "apply_spec_augment": true,
+  "apply_spec_augment": false,
   "architectures": [
     "WhisperForConditionalGeneration"
   ],
   "attention_dropout": 0.0,
-  "begin_suppress_tokens": null,
+  "begin_suppress_tokens": [
+    220,
+    50257
+  ],
   "bos_token_id": 50257,
-  "bpe_dropout": 0.05,
   "classifier_proj_size": 256,
   "d_model": 1280,
   "decoder_attention_heads": 20,
@@ -25,13 +27,13 @@
   "eos_token_id": 50257,
   "init_std": 0.02,
   "is_encoder_decoder": true,
-  "mask_feature_length": 14,
-  "mask_feature_min_masks": 2,
-  "mask_feature_prob": 0.22,
+  "mask_feature_length": 10,
+  "mask_feature_min_masks": 0,
+  "mask_feature_prob": 0.05,
   "mask_time_length": 10,
   "mask_time_min_masks": 2,
-  "mask_time_prob": 0.3,
-  "max_length": null,
+  "mask_time_prob": 0.05,
+  "max_length": 448,
   "max_source_positions": 1500,
   "max_target_positions": 448,
   "median_filter_width": 7,

generation_config.json

@@ -1,44 +1,84 @@
 {
   "alignment_heads": [
     [
-      7,
+      1,
       0
     ],
     [
-      10,
-      17
+      1,
+      1
     ],
     [
-      12,
-      18
+      1,
+      2
+    ],
+    [
+      1,
+      3
+    ],
+    [
+      1,
+      4
+    ],
+    [
+      1,
+      5
+    ],
+    [
+      1,
+      6
+    ],
+    [
+      1,
+      7
+    ],
+    [
+      1,
+      8
+    ],
+    [
+      1,
+      9
     ],
     [
-      13,
+      1,
+      10
+    ],
+    [
+      1,
+      11
+    ],
+    [
+      1,
       12
     ],
     [
-      16,
-      1
+      1,
+      13
     ],
     [
-      17,
+      1,
       14
     ],
     [
-      19,
-      11
+      1,
+      15
     ],
     [
-      21,
-      4
+      1,
+      16
     ],
     [
-      24,
-      1
+      1,
+      17
     ],
     [
-      25,
-      6
+      1,
+      18
+    ],
+    [
+      1,
+      19
     ]
   ],
   "begin_suppress_tokens": [
@@ -48,6 +88,16 @@
   "bos_token_id": 50257,
   "decoder_start_token_id": 50258,
   "eos_token_id": 50257,
+  "forced_decoder_ids": [
+    [
+      1,
+      50358
+    ],
+    [
+      2,
+      50360
+    ]
+  ],
   "is_multilingual": true,
   "lang_to_id": {
     "<|af|>": 50327,
@@ -151,6 +201,7 @@
     "<|yo|>": 50325,
     "<|zh|>": 50260
   },
+  "language": "auto",
   "max_initial_timestamp_index": 50,
   "max_length": 448,
   "no_timestamps_token_id": 50364,
@@ -247,6 +298,7 @@
     50362,
     50363
   ],
+  "task": "transcribe",
   "task_to_id": {
     "transcribe": 50360,
     "translate": 50359