Skip to content
/ whisper Public

Robust Speech Recognition via Large-Scale Weak Supervision

License

Notifications You must be signed in to change notification settings

openai/whisper

Repository files navigation

whisper-torchscript

See original readme and repo. This repo modifies Whisper a little bit to enable TorchScript.

What's new?

  • TorchScript-able model.
  • kv_cache will be passed normally instead of using hooks.
  • Cannot use the existing wrapper library with this new model code.
  • Some modules will be duplicated in favor of using less if elses.

Same as before:

  • Original checkpoints are still valid.
  • Model architecture is the same as before.