diff --git a/.gitignore b/.gitignore index 12379a0f..9cd9e383 100644 --- 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 index 89d09c9f..30a4cc38 100644 --- 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)