Spaces:
Build error
Build error
Adding support for large-v3 in faster-whisper
Browse files
src/whisper/fasterWhisperContainer.py
CHANGED
|
@@ -43,11 +43,11 @@ class FasterWhisperContainer(AbstractWhisperContainer):
|
|
| 43 |
model_url = model_config.url
|
| 44 |
|
| 45 |
if model_config.type == "whisper":
|
| 46 |
-
if model_url not in ["tiny", "base", "small", "medium", "large", "large-v1", "large-v2"]:
|
| 47 |
raise Exception("FasterWhisperContainer does not yet support Whisper models. Use ct2-transformers-converter to convert the model to a faster-whisper model.")
|
| 48 |
if model_url == "large":
|
| 49 |
-
# large is an alias for large-
|
| 50 |
-
model_url = "large-
|
| 51 |
|
| 52 |
device = self.device
|
| 53 |
|
|
|
|
| 43 |
model_url = model_config.url
|
| 44 |
|
| 45 |
if model_config.type == "whisper":
|
| 46 |
+
if model_url not in ["tiny", "base", "small", "medium", "large", "large-v1", "large-v2", "large-v3"]:
|
| 47 |
raise Exception("FasterWhisperContainer does not yet support Whisper models. Use ct2-transformers-converter to convert the model to a faster-whisper model.")
|
| 48 |
if model_url == "large":
|
| 49 |
+
# large is an alias for large-v3
|
| 50 |
+
model_url = "large-v3"
|
| 51 |
|
| 52 |
device = self.device
|
| 53 |
|