github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-25 14:58:46 -05:00
Code Issues Packages Projects Releases Wiki Activity
Files
02636ff62d95186dd541a245ef0df9cf418dedd2
tinygrad/examples/coder.py
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

4.6 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink