Laurent Mazare 329f661d9b Trace softmax (#568) ...

* Trace the softmax op.

* Inline the sum.

* Add min/max vec operations.

2023-08-23 15:25:50 +01:00

..

attention.rs

Trace softmax (#568)

2023-08-23 15:25:50 +01:00

clip.rs

Add a simple Module trait and implement it for the various nn layers (#500)

2023-08-18 09:38:22 +01:00

ddim.rs

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

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

embeddings.rs

Add a simple Module trait and implement it for the various nn layers (#500)

2023-08-18 09:38:22 +01:00

main.rs

Print some per-step timings in stable-diffusion. (#520)

2023-08-20 05:45:12 +01:00

resnet.rs

Add some group parameter to convolutions. (#566)

2023-08-23 12:58:55 +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

Add some group parameter to convolutions. (#566)

2023-08-23 12:58:55 +01:00

unet_2d.rs

Add some group parameter to convolutions. (#566)

2023-08-23 12:58:55 +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