feat: add support for new languages sk, sl, and hr

- Add languages 'sk', 'sl', and 'hr' to the GitHub Actions workflow. - Add languages 'sk', 'sl', and 'hr' to the Docker build target configuration. - Add align models for languages 'sk', 'sl', and 'hr' in `load_align_model.py`. Signed-off-by: 陳鈞 <jim60105@gmail.com>
jim60105 · Aug 25, 2024 · 0d77085 · 0d77085
1 parent cba5137
commit 0d77085
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/.github/workflows/docker_publish.yml b/.github/workflows/docker_publish.yml
@@ -243,6 +243,9 @@ jobs:
           # https://github.com/jim60105/docker-whisperX/actions/runs/8405597972
           # - no
           - nn
+          - sk
+          - sl
+          - hr
         model:
           - tiny
           - base

diff --git a/docker-bake.hcl b/docker-bake.hcl
@@ -51,6 +51,9 @@ target "build" {
       "ml",
       "no",
       "nn",
+      "sk",
+      "sl",
+      "hr"
     ]
   }
 

diff --git a/load_align_model.py b/load_align_model.py
@@ -39,6 +39,9 @@
     "ml": "gvs/wav2vec2-large-xlsr-malayalam",
     "no": "NbAiLab/nb-wav2vec2-1b-bokmaal",
     "nn": "NbAiLab/nb-wav2vec2-300m-nynorsk",
+    "sk": "comodoro/wav2vec2-xls-r-300m-sk-cv8",
+    "sl": "anton-l/wav2vec2-large-xlsr-53-slovenian",
+    "hr": "classla/wav2vec2-xls-r-parlaspeech-hr",
 }
 
 # From https://github.com/m-bain/whisperX/issues/189#issuecomment-1523392800
-Original file line number
+Diff line change
@@ Expand Up / @@ -243,6 +243,9 @@ jobs: @@
               # https://github.com/jim60105/docker-whisperX/actions/runs/8405597972
               # - no
               - nn
+              - sk
+              - sl
+              - hr
             model:
               - tiny
               - base
@@ Expand Down @@