chenyu f88506e630 move gpt2/llama sampling inside the model call (#3013) ...

* move gpt2/llama sampling inside the model call

* argmax uses one more kernel

2024-01-04 17:01:50 -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

Apply ruff linting rules to tests (#2473)

2023-11-27 21:24:06 -08:00

test_efficientnet.py

Apply ruff linting rules to tests (#2473)

2023-11-27 21:24:06 -08:00

test_end2end.py

call ruff with --preview (#2522)

2023-11-30 13:59:00 -05:00

test_mnist.py

Remove pytest markers (#2831)

2023-12-18 18:53:28 -05:00

test_onnx.py

fix some long lines in tests (#3006)

2024-01-03 23:53:33 -05:00

test_real_world.py

move gpt2/llama sampling inside the model call (#3013)

2024-01-04 17:01:50 -05:00

test_rnnt.py

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

test_train.py

Remove pytest markers (#2831)

2023-12-18 18:53:28 -05:00

test_waifu2x.py

Apply ruff linting rules to tests (#2473)

2023-11-27 21:24:06 -08:00

test_whisper.py

ruff checks the max line length is 150 (#2734)

2023-12-12 17:34:47 -08:00