fix: catch for IndexError

winstxnhdw · Jun 6, 2024 · 65c4529 · 65c4529
1 parent 25dc3e8
commit 65c4529
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/capgen/transcriber/transcriber.py b/capgen/transcriber/transcriber.py
@@ -50,7 +50,7 @@ def __init__(
         try:
             self.model = WhisperModel(**model_parameters, flash_attention=True)
 
-        except ValueError:
+        except (ValueError, IndexError):
             self.model = WhisperModel(**model_parameters)
 
     async def transcribe(self, file: str | BinaryIO, caption_format: str) -> str | None: