Merge pull request #310 from jhj0517/feature/add-trubo

Update available models
jhj0517 · Oct 2, 2024 · fd091ff · fd091ff
2 parents 8295934 + 7d4cc15
commit fd091ff
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -2,6 +2,8 @@
 *.png
 *.mp4
 *.mp3
+.idea/
+.pytest_cache/
 venv/
 modules/ui/__pycache__/
 outputs/

diff --git a/modules/whisper/faster_whisper_inference.py b/modules/whisper/faster_whisper_inference.py
@@ -159,7 +159,7 @@ def get_model_paths(self):
         ----------
         Name list of models
         """
-        model_paths = {model:model for model in whisper.available_models()}
+        model_paths = {model:model for model in faster_whisper.available_models()}
         faster_whisper_prefix = "models--Systran--faster-whisper-"
 
         existing_models = os.listdir(self.model_dir)
-Original file line number
+Diff line change
@@ Expand Up / @@ -2,6 +2,8 @@ @@
     *.png
     *.mp4
     *.mp3
+    .idea/
+    .pytest_cache/
     venv/
     modules/ui/__pycache__/
     outputs/
@@ Expand Down @@