Tobias FischerandGitHub f9e32f2bb2 clip device fix (#6924)

2024-10-07 00:47:32 +08:00

..

bert.py

dtype=float in bert log_softmax for loss and accuracy (#6916)

2024-10-06 11:15:56 -04:00

clip.py

clip device fix (#6924)

2024-10-07 00:47:32 +08:00

convnext.py

move to new cached fetch (#2493)

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

efficientnet.py

remove the magic methods for moving between devices [pr] (#6881)

2024-10-04 20:27:52 +08:00

inception.py

Compute FID Score (#6802)

2024-10-01 19:47:58 -04:00

llama.py

switch symbolic from old to uops, final PR (#6872)

2024-10-04 16:42:27 +08: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

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