Laurent Mazare 5d99026fd2 F16 support for stable diffusion (#488) ...

* F16 support for stable diffusion.

* Keep the attention bits in F32.

* Keep more of the attention bits in F32.

* More mixed precision support.

2023-08-17 13:48:56 +01:00

..

attention.rs

F16 support for stable diffusion (#488)

2023-08-17 13:48:56 +01:00

clip.rs

F16 support for stable diffusion (#488)

2023-08-17 13:48:56 +01:00

ddim.rs

Main diffusion loop for the SD example. (#332)

2023-08-06 21:39:53 +01:00

embeddings.rs

F16 support for stable diffusion (#488)

2023-08-17 13:48:56 +01:00

main.rs

F16 support for stable diffusion (#488)

2023-08-17 13:48:56 +01:00

resnet.rs

Track the conv2d operations in stable-diffusion. (#431)

2023-08-13 15:58:26 +01:00

schedulers.rs

Add the ddim scheduler. (#330)

2023-08-06 20:44:00 +01:00

stable_diffusion.rs

F16 support for stable diffusion (#488)

2023-08-17 13:48:56 +01:00

unet_2d_blocks.rs

Flash-attention support in stable diffusion (#487)

2023-08-17 12:16:40 +01:00

unet_2d.rs

F16 support for stable diffusion (#488)

2023-08-17 13:48:56 +01:00

utils.rs

Track the conv2d operations in stable-diffusion. (#431)

2023-08-13 15:58:26 +01:00

vae.rs

Add a stable diffusion example (#328)

2023-08-06 17:49:43 +01:00