From 0eae9e1f50d174e6f6f14c1ecfb92807ec1dc16c Mon Sep 17 00:00:00 2001 From: Mahmoud Ashraf <32404268+MahmoudAshraf97@users.noreply.github.com> Date: Wed, 1 Feb 2023 03:02:10 +0200 Subject: [PATCH] added several wav2vec2 models by jonatasgrosman since his models were used in other languages before and I tested the arabic model myself, I assumed it's safe to include all the available models --- whisperx/alignment.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/whisperx/alignment.py b/whisperx/alignment.py index ea97d8c..fd92d9e 100644 --- a/whisperx/alignment.py +++ b/whisperx/alignment.py @@ -30,6 +30,12 @@ DEFAULT_ALIGN_MODELS_HF = { "uk": "Yehor/wav2vec2-xls-r-300m-uk-with-small-lm", "pt": "jonatasgrosman/wav2vec2-large-xlsr-53-portuguese", "ar": "jonatasgrosman/wav2vec2-large-xlsr-53-arabic", + "ru": "jonatasgrosman/wav2vec2-large-xlsr-53-russian", + "pl": "jonatasgrosman/wav2vec2-large-xlsr-53-polish", + "hu": "jonatasgrosman/wav2vec2-large-xlsr-53-hungarian", + "fi": "jonatasgrosman/wav2vec2-large-xlsr-53-finnish", + "fa": "jonatasgrosman/wav2vec2-large-xlsr-53-persian", + "el": "jonatasgrosman/wav2vec2-large-xlsr-53-greek", }