support for nllb200 translation model #96

abdeladim-s · Jan 4, 2024 · e28c7ef · e28c7ef
1 parent 3a81bd8
commit e28c7ef
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -8,6 +8,6 @@ whisper-timestamped @ git+https://github.com/linto-ai/whisper-timestamped
 pandas==1.5.3
 pysubs2~=1.6.0
 pywhispercpp==1.1.1
-dl_translate==0.2.6
+dl_translate==0.3.0
 faster_whisper==0.10.0
 whisperx @ git+https://github.com/m-bain/whisperx@0ae0d49d1dd789322181f579c71719f4cac83ae1
diff --git a/src/subsai/utils.py b/src/subsai/utils.py
@@ -44,6 +44,7 @@ def available_translation_models() -> list:
         "facebook/m2m100_418M",
         "facebook/m2m100_1.2B",
         "facebook/mbart-large-50-many-to-many-mmt",
+        "facebook/nllb-200-distilled-600M"
     ]
     return models