3 people c1bbd15bd9 Sharded SDXL Inference (#6328) ...

* initial sharding fixes

* sigma device fix

* emptyline space fix

---------

Co-authored-by: chenyu <[email protected]>

2024-09-21 01:26:43 -04: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

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

Sharded SDXL Inference (#6328)

2024-09-21 01:26:43 -04:00

vit.py

move to new cached fetch (#2493)

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