Laurent Mazare 1c9e5394a5 Add a custom softmax implementation. (#744) ... * Add a custom softmax implementation. * Add softmaxlastdim to the benchmarks. * And add a test. * Support more dtypes. * Polish the code. * Use the slow implementation on cuda. * Add a todo for the cuda kernel.		2023-09-05 14:20:23 +01:00
..
attention.rs	Add a custom softmax implementation. (#744)	2023-09-05 14:20:23 +01:00
clip.rs	Preliminary support for SDXL. (#647)	2023-08-29 09:00:04 +01:00
ddim.rs	Remove some dead-code annotations. (#629)	2023-08-27 18:52:33 +01:00
embeddings.rs	Remove some dead-code annotations. (#629)	2023-08-27 18:52:33 +01:00
main.rs	Preliminary support for SDXL. (#647)	2023-08-29 09:00:04 +01:00
resnet.rs	Dilated convolutions (#657)	2023-08-29 16:12:11 +01:00
schedulers.rs	Add the ddim scheduler. (#330)	2023-08-06 20:44:00 +01:00
stable_diffusion.rs	Use multiple transformer layer in the same cross-attn blocks. (#653)	2023-08-29 11:13:43 +01:00
unet_2d.rs	Use multiple transformer layer in the same cross-attn blocks. (#653)	2023-08-29 11:13:43 +01:00
unet_2d_blocks.rs	Simplify usage of the pool functions. (#662)	2023-08-29 19:12:16 +01:00
utils.rs	dinov2 - read images from disk and compute the class probabilities (#503)	2023-08-18 15:50:33 +01:00
vae.rs	Add some group parameter to convolutions. (#566)	2023-08-23 12:58:55 +01:00