pytorch / audio

Data manipulation and transformation for audio signal processing, powered by PyTorch
https://pytorch.org/audio
BSD 2-Clause "Simplified" License
2.55k stars 657 forks source link

How to use my finetuned version of wave2vec2 for forced alignment as shown in example/ #3796

Open omerarshad opened 6 months ago

omerarshad commented 6 months ago

🐛 Describe the bug

Example script i am following, it used default pretrained model, where as. i want to use my own finetuned model.

https://pytorch.org/audio/main/generated/torchaudio.pipelines.Wav2Vec2FABundle.html#torchaudio.pipelines.Wav2Vec2FABundle

Versions

[pip3] mypy-extensions==1.0.0 [pip3] numpy==1.24.4 [pip3] onnx==1.15.0 [pip3] onnxruntime==1.16.3 [pip3] torch==2.2.2 [pip3] torchaudio==2.2.2 [pip3] torchvision==0.15.2 [conda] numpy 1.24.4 pypi_0 pypi [conda] torch 2.2.2 pypi_0 pypi [conda] torchaudio 2.2.2 pypi_0 pypi [conda] torchvision 0.15.2 pypi_0 pypi