egorsmkv/wav2vec2-jit

Export a traced JIT version of wav2vec2 models

Export a traced JIT version of wav2vec2 models

Install

pip install torch torchaudio transformers pyctcdecode

Use

python export_cpu_version.py

# or

python export_gpu_version.py

Usage without a language model

python recognize_cpu.py

# or

python recognize_gpu.py

Usage with a language model

python recognize_cpu_with_lm.py

# or

python recognize_gpu_with_lm.py