mirror of
https://github.com/openai/whisper.git
synced 2025-03-30 14:28:27 +00:00
500d0fe966
* 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>
15 lines
214 B
Python
15 lines
214 B
Python
import random as rand
|
|
|
|
import numpy
|
|
import pytest
|
|
|
|
|
|
def pytest_configure(config):
|
|
config.addinivalue_line("markers", "requires_cuda")
|
|
|
|
|
|
@pytest.fixture
|
|
def random():
|
|
rand.seed(42)
|
|
numpy.random.seed(42)
|