feat: Add align model hi

jim60105 · Aug 29, 2023 · 78b37de · 78b37de
1 parent 5e67c0d
commit 78b37de
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 1 deletion.
diff --git a/.github/workflows/docker_publish.yml b/.github/workflows/docker_publish.yml
@@ -116,6 +116,7 @@ jobs:
           - ko
           - ur
           - te
+          - hi
         model:
           - tiny
           - base

diff --git a/.vscode/settings.json b/.vscode/settings.json
@@ -11,6 +11,7 @@
         "mpoyraz",
         "nguyenvulebinh",
         "saattrupdan",
+        "theainerd",
         "torchaudio",
         "VOXPOPULI",
         "whisperx",

diff --git a/docker-bake.hcl b/docker-bake.hcl
@@ -44,7 +44,8 @@ target "build" {
       "vi",
       "ko",
       "ur",
-      "te"
+      "te",
+      "hi",
     ]
   }
 

diff --git a/load_align_model.py b/load_align_model.py
@@ -34,6 +34,7 @@
     "ko": "kresnik/wav2vec2-large-xlsr-korean",
     "ur": "kingabzpro/wav2vec2-large-xls-r-300m-Urdu",
     "te": "anuragshas/wav2vec2-large-xlsr-53-telugu",
+    "hi": "theainerd/Wav2Vec2-large-xlsr-hindi",
 }
 
 # From https://github.com/m-bain/whisperX/issues/189#issuecomment-1523392800
-Original file line number
+Diff line change
@@ Expand Up / @@ -116,6 +116,7 @@ jobs: @@
               - ko
               - ur
               - te
+              - hi
             model:
               - tiny
               - base
@@ Expand Down @@