tinygrad/test/models
geohotstan 1e904155e3
Add Onnx Huggingface to test/models/test_onnx.py (#11468)
* BOOM

* cache extra/huggingface/models/

* why max buffer size is not 0

* override MAX_BUFFER_SIZE

* less models

* remove more models and change cache dir to already cached dir

* only metal

* less is more?

* remove check ops

* why is this not setting the ENVVAR

* ughhhhh just test in models

* only cpu and gpu

* only cpu actually

* just override it idk

* final

* move extra dependencies up top

* simplification

* fix print

* make README better

* revert ops_disk fix for now

* clean up test_onnx

* remove testing fashion clip model cuz sloooowwwwww

* actually let METAL run this

* fix comment mistake

* fix download path in run_models

* does this work?

* cleanup setup and teardown

* contextvar like this?

* prove model is cached

* do I need to increment DOWNLOAD_CACHE_VERSION?

* see if cached with incremented DOWNLOAD_CACHE_VERSION

* use warnings to see if the model exists

* revert DOWNLOAD_CACHE_VERSION stuff and clean up

* add retry to download

* nit
2025-08-14 11:16:41 -04: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 Apply ruff linting rules to tests (#2473) 2023-11-27 21:24:06 -08:00
test_efficientnet.py db in wal mode (#5388) 2024-07-12 20:43:36 -07:00
test_end2end.py tests from grad uop path [pr] (#8313) 2024-12-18 09:25:05 -08:00
test_mnist.py CLANG -> CPU (#9189) 2025-02-20 18:03:09 -05:00
test_onnx.py Add Onnx Huggingface to test/models/test_onnx.py (#11468) 2025-08-14 11:16:41 -04:00
test_real_world.py clean up binary_crossentropy_logits (#10958) 2025-06-24 12:23:40 -04:00
test_resnet.py Fix FC layer ResNet load_from_pretrained error (#8387) 2024-12-26 18:11:27 -05:00
test_rnnt.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
test_train.py hotfix: those tests fail locally on mac due to buffer count 2025-01-27 07:53:48 +09:00
test_waifu2x.py Apply ruff linting rules to tests (#2473) 2023-11-27 21:24:06 -08:00
test_whisper.py Fix whisper OOB (#10685) 2025-06-07 20:23:50 -04:00