Laurent Mazare dcf708559d Fix for cudnn to work with img2img. (#753)

2023-09-06 07:49:28 +01:00

..

attention.rs

Add a custom softmax implementation. (#744)

2023-09-05 14:20:23 +01:00

clip.rs

Preliminary support for SDXL. (#647)

2023-08-29 09:00:04 +01:00

ddim.rs

img2img pipeline for stable diffusion. (#752)

2023-09-06 07:06:49 +01:00

embeddings.rs

Remove some dead-code annotations. (#629)

2023-08-27 18:52:33 +01:00

main.rs

img2img pipeline for stable diffusion. (#752)

2023-09-06 07:06:49 +01:00

resnet.rs

Dilated convolutions (#657)

2023-08-29 16:12:11 +01:00

schedulers.rs

Add the ddim scheduler. (#330)

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

stable_diffusion.rs

Use multiple transformer layer in the same cross-attn blocks. (#653)

2023-08-29 11:13:43 +01:00

unet_2d_blocks.rs

Fix for cudnn to work with img2img. (#753)

2023-09-06 07:49:28 +01:00

unet_2d.rs

Use multiple transformer layer in the same cross-attn blocks. (#653)

2023-08-29 11:13:43 +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