update config

Signed-off-by: nithinraok <[email protected]>

.gitattributes

@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 parakeet-ctc-1.1b.nemo filter=lfs diff=lfs merge=lfs -text
+.safetensors filter=lfs diff=lfs merge=lfs -text

config.json

@@ -1,6 +1,6 @@
 {
   "architectures": [
-    "FastConformerForCTC"
+    "ParakeetCTC"
   ],
   "blank_token_id": 0,
   "bos_token_id": 1,
@@ -11,7 +11,7 @@
     "activation_dropout": 0.1,
     "activation_function": "silu",
     "architectures": [
-      "FastConformerModel"
+      "ParakeetCTC"
     ],
     "attention_bias": false,
     "attention_dropout": 0.1,
@@ -35,7 +35,7 @@
     "nemo_decoder_type": "nemo.collections.asr.modules.ConvASRDecoder",
     "nemo_encoder_type": "nemo.collections.asr.modules.ConformerEncoder",
     "nemo_model_name": "nvidia/parakeet-ctc-1.1b",
-    "nemo_model_type": "parakeet",
+    "nemo_model_type": "parakeet_ctc",
     "num_attention_heads": 8,
     "num_hidden_layers": 42,
     "num_key_value_heads": 8,
@@ -51,7 +51,7 @@
     "vocab_size": 1024,
     "xscaling": true
   },
-  "model_type": "fastconformer_for_ctc",
+  "model_type": "parakeet_ctc",
   "pad_token_id": 0,
   "torch_dtype": "float32",
   "transformers_version": "4.53.0.dev0",

preprocessor_config.json

@@ -5,14 +5,13 @@
   "feature_size": 80,
   "hop_length": 160,
   "mag_power": 2.0,
-  "mel_norm": "slaney",
-  "mel_scale": "htk",
+  "mel_scale": "slaney",
   "n_fft": 512,
   "n_mels": 80,
   "normalize": "per_feature",
   "padding_side": "right",
   "padding_value": 0.0,
-  "preemph": 0.97,
+  "preemphasis": 0.97,
   "return_attention_mask": true,
   "sampling_rate": 16000,
   "win_length": 400,