whisper/requirements.txt

numba
numpy
torch
tqdm
more-itertools
tiktoken
triton>=2.0.0;platform_machine=="x86_64" and sys_platform=="linux" or sys_platform=="linux2"
word-level timestamps in `transcribe()` (#869) * word-level timestamps in `transcribe()` * moving to `timing.py` * numba implementation for dtw, replacing dtw-python * triton implementation for dtw * add test for dtw implementations * triton implementation of median_filter * a simple word-level timestamps test * add scipy as dev dependency * installs an older version of Triton if CUDA < 11.4 * fix broken merge * loosen nvcc version match regex * find_alignment() function * miscellaneous improvements * skip median filtering when the input is too small * Expose punctuation options in cli and transcribe() (#973) * fix merge error * fix merge error 2 * annotating that word_timestamps is experimental --------- Co-authored-by: ryanheise <ryan@ryanheise.com> 2023-03-06 22:00:49 +00:00			`numba`
initial commit 2022-09-21 15:43:20 +00:00			`numpy`
			`torch`
			`tqdm`
Add conda environment.yml (and fix requirements.txt) (#8) * fix: more-itertools name in requirements.txt * feature: minimal environment.yml for conda * Revert "feature: minimal environment.yml for conda" This reverts commit 8fd7438b368b0eb5df85f667fea911f293fa5e6d. Co-authored-by: Jong Wook Kim <jongwook@nyu.edu> 2022-09-23 03:30:45 +00:00			`more-itertools`
remove tiktoken pin (#1759) 2023-11-06 11:05:21 +00:00			`tiktoken`
Relax triton requirements for compatibility with pytorch 2.4 and newer (#2307) * Relax triton requirements for compatibility with pytorch 2.4 and newer Similar to https://github.com/openai/whisper/pull/1802, but now when pytorch upgrades to 2.4, it requires triton==3.0.0. I am not sure if it makes sense to remove the upper bound version constraints * Update requirements.txt 2024-09-10 16:53:08 +00:00			`triton>=2.0.0;platform_machine=="x86_64" and sys_platform=="linux" or sys_platform=="linux2"`