tinygrad/extra/models
Tobias Fischer 0c3a35e5c2
Stable Diffusion v2 Inference (#5283)
* model implementation

* clip fix, more qol options
2024-07-03 22:47:10 -04:00
..
bert.py Residual in MLM loss + Change default steps (#4935) 2024-06-12 16:09:18 -04:00
clip.py Stable Diffusion v2 Inference (#5283) 2024-07-03 22:47:10 -04: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 nn.RMSNorm (#5272) 2024-07-02 21:39:01 -04:00
mask_rcnn.py remove numpy from dtype (#4969) 2024-06-14 15:38:45 -04:00
resnet.py Fix batchnorm shapes for resnet.load_pretrained (#5167) 2024-06-26 18:44:10 -04:00
retinanet.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00
rnnt.py change Tensor.stack to method (#4719) 2024-05-24 17:04:19 -04:00
transformer.py replace with tensor op (#3099) 2024-01-12 14:13:40 -05:00
unet.py Pulled CLIP and UNet into Seperate Files (#5253) 2024-07-01 22:33:01 -04: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