tinygrad/test/models
C T 26f8b12e01
Whisper audio helpers (mel filters in tinygrad) (#13478)
* add whisper audio helpers for stft/mel/resample

* cleanup

* add whisper stft test

* make only stft test explicitly depend on librosa

* extract sinc_window_kernel

* dehardcode device

* use same device argument

* simplify

* type annotate

* ruff format audio_helpers.py

* ruff format test_whisper.py

* add WHISPER_NEW_STFT

* rename

* undo ruff format changes

* use new stft and mel for whisper

* remove stft test that depends on librosa

* remove whitespace

* add Tensor.log10 with test\test_ops.py::TestOps::test_log10

* use Tensor.log10

* fix lint

* future: remove unused STFT class

* future: remove resample code since it isn't used (yet)

* match openai with pad_mode="reflect"

* pad_to

* future: cut resample leftovers

* cleanup

* add mel tests

* future: cut stft

* future: cut non-mel prep_audio changes

* reduce diff

* move audio_helpers.py to examples

* reduce whitespace

* fix imports

* reduce whitespace

---------

Co-authored-by: chenyu <chenyu@fastmail.com>
2026-01-20 10:50:02 -05:00
..
efficientnet test cleanups 2023-03-03 06:36:06 -08:00
waifu2x waifu2x vgg7: testcase, auto-RGBA->RGB, function to grab pretrained models, training "fix" (#2117) 2023-10-19 22:07:15 -07:00
whisper whisper: support batch inference, add librispeech WER test (#2074) 2023-11-16 13:50:08 -08:00
__init__.py more test cleanups (#2631) 2023-12-05 16:17:57 -08:00
test_bert.py update tests, WEBGPU=1 pytest . passes (#14089) 2026-01-10 00:03:02 -05:00
test_efficientnet.py fix tests that failed locally on mac (#13872) 2025-12-29 11:23:38 -05:00
test_end2end.py add needs_second_gpu decorator (#13543) 2025-12-02 19:08:23 -08:00
test_mnist.py mark slow tests as slow instead of as CI (#13736) 2025-12-17 10:29:57 -04:00
test_onnx.py add Swish op (#13541) 2025-12-08 12:41:18 -05:00
test_real_world.py mark slow tests as slow instead of as CI (#13736) 2025-12-17 10:29:57 -04:00
test_resnet.py update tests, WEBGPU=1 pytest . passes (#14089) 2026-01-10 00:03:02 -05:00
test_rnnt.py update test/models (#12142) 2025-09-12 16:43:28 -04:00
test_train.py mark slow tests as slow instead of as CI (#13736) 2025-12-17 10:29:57 -04:00
test_waifu2x.py Apply ruff linting rules to tests (#2473) 2023-11-27 21:24:06 -08:00
test_whisper.py Whisper audio helpers (mel filters in tinygrad) (#13478) 2026-01-20 10:50:02 -05:00