Fix model type

README.md

@@ -44,9 +44,6 @@ model = nemo_asr.models.ASRModel.from_pretrained("HiTZ/stt_eu_conformer_transduc
 onnx_dir = Path("onnx-dir")
 onnx_dir.mkdir(exist_ok=True)
 
-if hasattr(model, "decoder") and hasattr(model.decoder, "vocab_size"):
-    model.decoder.vocab_size += 1  # to account for <blk>
-
 model.export(str(Path(onnx_dir, "model.onnx")))
 
 with Path(onnx_dir, "vocab.txt").open("wt") as f:

config.json

@@ -1 +1 @@
-{"model_type": "nemo-conformer-tdt", "features_size": 80, "subsampling_factor": 4, "max_tokens_per_step": 10}
+{"model_type": "nemo-conformer-rnnt", "features_size": 80, "subsampling_factor": 4, "max_tokens_per_step": 10}