Laurent Mazare 2345b8ce3f Skeleton for the avg-pool2d and upsample-nearest2d ops. (#337) ...

* Skeleton for the avg-pool2d and upsample-nearest2d ops.

* Preliminary conv2d support.

2023-08-07 16:15:38 +01:00

..

attention.rs

Add a stable diffusion example (#328)

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

clip.rs

Implement group-norm. (#334)

2023-08-07 06:53:05 +01:00

ddim.rs

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

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

embeddings.rs

Simple pad support. (#336)

2023-08-07 15:24:56 +01:00

main.rs

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

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

resnet.rs

Add a stable diffusion example (#328)

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

schedulers.rs

Add the ddim scheduler. (#330)

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

stable_diffusion.rs

Add the ddim scheduler. (#330)

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

unet_2d_blocks.rs

Skeleton for the avg-pool2d and upsample-nearest2d ops. (#337)

2023-08-07 16:15:38 +01:00

unet_2d.rs

Add a stable diffusion example (#328)

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

utils.rs

Skeleton for the avg-pool2d and upsample-nearest2d ops. (#337)

2023-08-07 16:15:38 +01:00

vae.rs

Add a stable diffusion example (#328)

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