Laurent Mazare 9af438ac1b Track the conv2d operations in stable-diffusion. (#431) ...

* Track the conv2d operations in stable-diffusion.

* Add more tracing to stable-diffusion.

* Also trace the resnet bits.

* Trace the attention blocks.

* Also trace the attention inner part.

* Small tweak.

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

..

attention.rs

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

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

clip.rs

Fix the padding used in stable diffusion. (#362)

2023-08-09 13:23:59 +01:00

ddim.rs

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

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

embeddings.rs

Fixes for the stable diffusion example. (#342)

2023-08-08 14:57:09 +01:00

main.rs

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

2023-08-13 15:58:26 +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

Stable diffusion: retrieve the model files from the HF hub. (#414)

2023-08-11 18:57:06 +01:00

unet_2d_blocks.rs

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

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

unet_2d.rs

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

2023-08-13 15:58:26 +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