tinygrad/extra/models
samm393 19c11792fd
Flux.1 (#6334)
* initial commit

* whitespace

* get rid of torch import

* indentation

* less hardcoding

* add flux.1-dev

* jit

* no double

* t5 tidy up

* validation image

* reuse sdxl autoencoder

* typing changes

* empty lines

* remove unneeded comments

---------

Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
2024-09-24 10:08:04 +08:00
..
bert.py update pylint path to check indent/space for all (#6022) 2024-08-10 14:41:09 -04:00
clip.py sdxl batched inference fixes (#6293) 2024-08-28 07:44:58 -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
inception.py FID Inception Model (#5516) 2024-07-16 23:12:03 -04:00
llama.py faster tinychat (#5993) 2024-08-08 19:16:26 -07:00
mask_rcnn.py remove numpy from dtype (#4969) 2024-06-14 15:38:45 -04:00
resnet.py hotfix: resnet to obj.device 2024-09-06 13:06:02 +08: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
t5.py Flux.1 (#6334) 2024-09-24 10:08:04 +08:00
transformer.py replace with tensor op (#3099) 2024-01-12 14:13:40 -05:00
unet.py Sharded SDXL Inference (#6328) 2024-09-21 01:26:43 -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