aadnk commited on
Commit
4749691
1 Parent(s): 4855017

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-v1
50
- model_url = "large-v1"
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