tinygrad/test/models
mmmkkaaayy 8235da11dd
whisper: support batch inference, add librispeech WER test (#2074)
* whisper: support batch inference, add librispeech WER test, add kv caching and JIT

* remove JIT_SUPPORTED_DEVICE

---------

Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
2023-11-16 13:50:08 -08: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
test_bert.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
test_efficientnet.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
test_end2end.py clean up symlinking in benchmark (#2219) 2023-11-05 16:46:05 -08:00
test_mnist.py move state to nn/state (#1619) 2023-08-22 07:36:24 -07:00
test_onnx.py update openpilot supercombo.onnx to 0.9.4 (#1681) 2023-08-26 19:16:08 -04:00
test_real_world.py JIT support in Interpreted (#2314) 2023-11-15 11:13:38 -08:00
test_rnnt.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
test_train.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
test_waifu2x.py waifu2x vgg7: testcase, auto-RGBA->RGB, function to grab pretrained models, training "fix" (#2117) 2023-10-19 22:07:15 -07:00
test_whisper.py whisper: support batch inference, add librispeech WER test (#2074) 2023-11-16 13:50:08 -08:00