Tobias Fischer 85d4ca7caa FID Inception Model (#5516) ...

* added model impl

* minor cleanups

* extracted weights loading into from_pretrained

* reorganized model for better weight loading

* removed lru cache for state dict loading

2024-07-16 23:12:03 -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

inception.py

FID Inception Model (#5516)

2024-07-16 23:12:03 -04:00

llama.py

bring tinychat more inline with tinyos' version (#5358)

2024-07-10 13:13:52 -07: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

unet3d.py

move to new cached fetch (#2493)

2023-11-28 17:36:55 -08:00

unet.py

Pulled CLIP and UNet into Seperate Files (#5253)

2024-07-01 22:33:01 -04:00

vit.py

move to new cached fetch (#2493)

2023-11-28 17:36:55 -08:00