| 157 | return WHISPER_MODEL_SIZES.get(size, None) |
| 158 | |
| 159 | class ModelType(enum.Enum): |
| 160 | WHISPER = "Whisper" |
| 161 | WHISPER_CPP = "Whisper.cpp" |
| 162 | HUGGING_FACE = "Hugging Face" |
| 163 | FASTER_WHISPER = "Faster Whisper" |
| 164 | OPEN_AI_WHISPER_API = "OpenAI Whisper API" |
| 165 | |
| 166 | @property |
| 167 | def supports_initial_prompt(self): |
| 168 | return self in ( |
| 169 | ModelType.WHISPER, |
| 170 | ModelType.WHISPER_CPP, |
| 171 | ModelType.OPEN_AI_WHISPER_API, |
| 172 | ModelType.FASTER_WHISPER, |
| 173 | ) |
| 174 | |
| 175 | def is_available(self): |
| 176 | if ( |
| 177 | # Hide Faster Whisper option on macOS x86_64 |
| 178 | # See: https://github.com/SYSTRAN/faster-whisper/issues/541 |
| 179 | (self == ModelType.FASTER_WHISPER |
| 180 | and platform.system() == "Darwin" and platform.machine() == "x86_64") |
| 181 | ): |
| 182 | return False |
| 183 | return True |
| 184 | |
| 185 | def is_manually_downloadable(self): |
| 186 | return self in ( |
| 187 | ModelType.WHISPER, |
| 188 | ModelType.WHISPER_CPP, |
| 189 | ModelType.FASTER_WHISPER, |
| 190 | ) |
| 191 | |
| 192 | |
| 193 | HUGGING_FACE_MODEL_ALLOW_PATTERNS = [ |
no outgoing calls
no test coverage detected