Spaces:
Running
Running
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 |
|