github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity
Files
c0f76ed4ea3a6d75cbae0315fd2e8d2408d3d255
tinygrad/extra/models/llama.py
chenyu c0f76ed4ea transformer kvcache and mask have same dtype as input (#2771) 
* transformer kvcache and mask have same dtype as input

* don't use `=0` in cstyle ternary where

* (bool)

* where float16 test
2023-12-14 22:41:51 -05:00

7.9 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink