huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
26fd37b34854bd2b6bab97309d3ea9585921189f
candle/candle-examples/examples/stable-diffusion
History
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