├── .gitignore ├── MANIFEST.in ├── README.md ├── bmwhisper ├── __init__.py ├── __main__.py ├── assets │ ├── gpt2.tiktoken │ ├── mel_filters.npz │ ├── multilingual.tiktoken │ ├── positional_embedding_base.npz │ ├── positional_embedding_large-v2.npz │ ├── positional_embedding_large.npz │ ├── positional_embedding_medium.npz │ ├── positional_embedding_small.npz │ └── positional_embedding_tiny.npz ├── audio.py ├── decoding.py ├── model.py ├── model_export.py ├── normalizers │ ├── __init__.py │ ├── basic.py │ ├── english.json │ └── english.py ├── timing.py ├── tokenizer.py ├── transcribe.py ├── triton_ops.py ├── utils.py └── version.py ├── demo.wav ├── gen_bmodel.sh ├── gen_onnx.sh ├── requirements.txt ├── run.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/.gitignore -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/README.md -------------------------------------------------------------------------------- /bmwhisper/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/__init__.py -------------------------------------------------------------------------------- /bmwhisper/__main__.py: -------------------------------------------------------------------------------- 1 | from .transcribe import cli 2 | 3 | cli() 4 | -------------------------------------------------------------------------------- /bmwhisper/assets/gpt2.tiktoken: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/assets/gpt2.tiktoken -------------------------------------------------------------------------------- /bmwhisper/assets/mel_filters.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/assets/mel_filters.npz -------------------------------------------------------------------------------- /bmwhisper/assets/multilingual.tiktoken: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/assets/multilingual.tiktoken -------------------------------------------------------------------------------- /bmwhisper/assets/positional_embedding_base.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/assets/positional_embedding_base.npz -------------------------------------------------------------------------------- /bmwhisper/assets/positional_embedding_large-v2.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/assets/positional_embedding_large-v2.npz -------------------------------------------------------------------------------- /bmwhisper/assets/positional_embedding_large.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/assets/positional_embedding_large.npz -------------------------------------------------------------------------------- /bmwhisper/assets/positional_embedding_medium.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/assets/positional_embedding_medium.npz -------------------------------------------------------------------------------- /bmwhisper/assets/positional_embedding_small.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/assets/positional_embedding_small.npz -------------------------------------------------------------------------------- /bmwhisper/assets/positional_embedding_tiny.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/assets/positional_embedding_tiny.npz -------------------------------------------------------------------------------- /bmwhisper/audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/audio.py -------------------------------------------------------------------------------- /bmwhisper/decoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/decoding.py -------------------------------------------------------------------------------- /bmwhisper/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/model.py -------------------------------------------------------------------------------- /bmwhisper/model_export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/model_export.py -------------------------------------------------------------------------------- /bmwhisper/normalizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/normalizers/__init__.py -------------------------------------------------------------------------------- /bmwhisper/normalizers/basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/normalizers/basic.py -------------------------------------------------------------------------------- /bmwhisper/normalizers/english.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/normalizers/english.json -------------------------------------------------------------------------------- /bmwhisper/normalizers/english.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/normalizers/english.py -------------------------------------------------------------------------------- /bmwhisper/timing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/timing.py -------------------------------------------------------------------------------- /bmwhisper/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/tokenizer.py -------------------------------------------------------------------------------- /bmwhisper/transcribe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/transcribe.py -------------------------------------------------------------------------------- /bmwhisper/triton_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/triton_ops.py -------------------------------------------------------------------------------- /bmwhisper/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/bmwhisper/utils.py -------------------------------------------------------------------------------- /bmwhisper/version.py: -------------------------------------------------------------------------------- 1 | __version__ = "20230810" 2 | -------------------------------------------------------------------------------- /demo.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/demo.wav -------------------------------------------------------------------------------- /gen_bmodel.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/gen_bmodel.sh -------------------------------------------------------------------------------- /gen_onnx.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/gen_onnx.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- 1 | from bmwhisper.transcribe import * 2 | 3 | cli() 4 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JKay0327/whisper-TPU_py/HEAD/setup.py --------------------------------------------------------------------------------