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_blocks.rs

Simplify usage of the pool functions. (#662)

2023-08-29 19:12:16 +01:00

unet_2d.rs

Use multiple transformer layer in the same cross-attn blocks. (#653)

2023-08-29 11:13:43 +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