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

* Use multiple transformer layer in the same cross-attn blocks.

* Make the context contiguous if required.

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

..

attention.rs

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

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

clip.rs

Preliminary support for SDXL. (#647)

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

ddim.rs

Remove some dead-code annotations. (#629)

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

embeddings.rs

Remove some dead-code annotations. (#629)

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

main.rs

Preliminary support for SDXL. (#647)

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

resnet.rs

Remove some dead-code annotations. (#629)

2023-08-27 18:52:33 +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

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

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