OpenNMT / OpenNMT/CTranslate2

Request for Implementing Support for wav2vec2, MMS, and XLS-R Models

Open
#1,361 1 comment 3 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

enhancement
Dominant language
C++
Stars
4.7k
Forks
536
Avg merge
12h 12m
Merged PRs (30d)
4

Description

Hi there,

I really enjoy using this library and appreciate all your hard work. I noticed that you've added Whisper, which is great. But I've been working with other models like wav2vec2, MMS, and XLS-R and they perform really well.

What's great about them is that they need very little finetuning to get them to work really well, especially when we talk about WER. I've found Whisper needs a bit more finetuning in this regard, especially for languages that don't have a lot of resources.

I was wondering, do you plan to add support for these models, wav2vec2, MMS, and XLS-R, to the library? If not, could you guide me on how I might add them?

Here are some helpful links to know more about these models:

I think adding these models would make the library even better, especially for those who work with less commonly used languages.

Thanks a lot!

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by reviewing how the library currently supports Whisper, then read the linked MMS, XLSR-Wav2Vec2, and XLS-R model documentation. Determine the integration scope for each model and define what successful support and validation would require; the issue does not name files, entry points, or tests.

Written by the indexing model from the issue text.

Assessment

Tech stack
cpp
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.