tinygrad/extra/models
chenyu 5ac1fa933f
apply the same fix_bf16 in llama and coder (#3789)
* apply the same fix_bf16 in llama and coder

did not realize the same logic was in llama too.
really fix #2775

* flag for native SUPPORT_BF16 cast
2024-03-17 21:25:24 -04:00
..
bert.py multilazybuffer fix (#3609) 2024-03-04 17:36:23 -05:00
convnext.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00
efficientnet.py simple LoadOps.ASSIGN (#3745) 2024-03-14 20:44:34 -07:00
llama.py apply the same fix_bf16 in llama and coder (#3789) 2024-03-17 21:25:24 -04:00
mask_rcnn.py fix example that calls Tensor.__bool__ (#3650) 2024-03-07 16:59:26 -05:00
resnet.py MLPerf Resnet (cleaned up) (#3573) 2024-03-14 00:53:41 -04:00
retinanet.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00
rnnt.py move graph.py and jit.py into features (#3376) 2024-02-12 17:34:34 +01:00
transformer.py replace with tensor op (#3099) 2024-01-12 14:13:40 -05:00
unet3d.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00
vit.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00