nb-wav2vec2-1b-bokmaal-onnx (ONNX)

ONNX export of NbAiLab/nb-wav2vec2-1b-bokmaal, a Norwegian Bokmål wav2vec2 CTC ASR model from National Library of Norway AI Lab (NbAiLab), for use with onnx-asr (wav2vec2-ctc model type).

Per-utterance zero-mean/unit-variance normalization is baked into the ONNX graph, masked by input_lengths for correct behavior with padded/batched input, so the model works with onnx-asr's plain identity preprocessor (raw 16kHz waveform in).

Usage

import onnx_asr

model = onnx_asr.load_model("OpenVoiceOS/nb-wav2vec2-1b-bokmaal-onnx")
print(model.recognize("test.wav"))

Files

  • model.onnx / model.onnx.data — fp32 ONNX graph (inputs: input_values (batch, samples) float32, input_lengths (batch,) int64; output: logprobs (batch, frames, vocab) float32 log-softmax).
  • vocab.txt — CTC vocabulary in onnx-asr's token id format (word-delimiter -> , pad token -> <blk>).
  • config.json{"model_type": "wav2vec2-ctc", "subsampling_factor": 320}.

No int8 quantized variant is included yet -- onnxruntime.quantization does not currently support the torch.onnx dynamo-exported graph for this architecture.

License

Apache License 2.0, inherited from the base model.

Downloads last month
17
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for OpenVoiceOS/nb-wav2vec2-1b-bokmaal-onnx

Quantized
(1)
this model