4 ms·
I've been looking for faster implementations of Whisper, the main drawback with Whisper Jax is that the performance comes from running on Google TPUs, which are
by sebnun 3y ago
I've been looking for faster implementations of Whisper, the main drawback with Whisper Jax is that the performance comes from running on Google TPUs, which are much more expensive than GPUs.
On "normal" GPUs the fastest implementation I've found is https://github.com/guillaumekln/faster-whisper https://github.com/guillaumekln/faster-whisper. Whisper.cpp works faster on a CPU, especially on Apple Silicon, but still nowhere near the performance you could get on a GPU (understandably).
How does Whisper Jax compares to faster-whisper on a GPU?
- ttt3ts 3y agoYea, whisper runs fairly quickly on really old GPUs and leveraging CTranslate2 (underlying tech in your link) you need < 4GB of GPU RAM to run the large model. This might be faster but an order of magnitude more expensive? Also, huge lockin to one provider.
- zznzz 3y agojax and whisper-jax support running on GPUs in addition to TPUs, this particular demo just runs on a TPU.