s3://sagemaker-whisper-testing/output/base/en/sagemaker-whispertra-250719-0535-009-335b8026/output/model.tar.gz

config.json

@@ -13,17 +13,17 @@
   ],
   "bos_token_id": 50257,
   "classifier_proj_size": 256,
-  "d_model": 768,
-  "decoder_attention_heads": 12,
-  "decoder_ffn_dim": 3072,
+  "d_model": 512,
+  "decoder_attention_heads": 8,
+  "decoder_ffn_dim": 2048,
   "decoder_layerdrop": 0.0,
-  "decoder_layers": 12,
+  "decoder_layers": 6,
   "decoder_start_token_id": 50257,
-  "dropout": 0.014221097611095367,
-  "encoder_attention_heads": 12,
-  "encoder_ffn_dim": 3072,
+  "dropout": 0.005637927715008337,
+  "encoder_attention_heads": 8,
+  "encoder_ffn_dim": 2048,
   "encoder_layerdrop": 0.0,
-  "encoder_layers": 12,
+  "encoder_layers": 6,
   "eos_token_id": 50256,
   "forced_decoder_ids": null,
   "init_std": 0.02,
@@ -39,7 +39,7 @@
   "max_target_positions": 448,
   "median_filter_width": 7,
   "model_type": "whisper",
-  "num_hidden_layers": 12,
+  "num_hidden_layers": 6,
   "num_mel_bins": 80,
   "pad_token_id": 50256,
   "scale_embedding": false,
@@ -47,5 +47,5 @@
   "transformers_version": "4.45.2",
   "use_cache": false,
   "use_weighted_layer_sum": false,
-  "vocab_size": 51869
+  "vocab_size": 51867
 }

generation_config.json

@@ -1,80 +1,24 @@
 {
   "alignment_heads": [
     [
-      6,
-      6
-    ],
-    [
-      7,
-      0
-    ],
-    [
-      7,
+      3,
       3
     ],
     [
-      7,
-      8
-    ],
-    [
-      8,
-      2
-    ],
-    [
-      8,
-      5
-    ],
-    [
-      8,
+      4,
       7
     ],
     [
-      9,
-      0
-    ],
-    [
-      9,
-      4
-    ],
-    [
-      9,
-      8
-    ],
-    [
-      9,
-      10
-    ],
-    [
-      10,
-      0
-    ],
-    [
-      10,
+      5,
       1
     ],
     [
-      10,
-      2
-    ],
-    [
-      10,
-      3
-    ],
-    [
-      10,
-      6
-    ],
-    [
-      10,
-      11
-    ],
-    [
-      11,
-      2
+      5,
+      5
     ],
     [
-      11,
-      4
+      5,
+      7
     ]
   ],
   "begin_suppress_tokens": [
@@ -96,6 +40,7 @@
   "no_timestamps_token_id": 50362,
   "pad_token_id": 50256,
   "prev_sot_token_id": 50360,
+  "return_timestamps": false,
   "suppress_tokens": [],
   "transformers_version": "4.45.2"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4f5f0cc1ff7933348c1407e177adff55828a68aa562bcfc58dbabdec04743d3b
-size 967007368
+oid sha256:3cdbd67a72b9653850afc5e2cc9d35849d30cf1b6eb719ed33703944005329c6
+size 290408032