github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-06 20:55:17 -05:00
Code Issues Packages Projects Releases Wiki Activity
Files
73b72b8de2fa9b63023c1e67c19c93a7f24b480c
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