Laurent Mazare 2d3fcad267 Simplify usage of the pool functions. (#662) ...

* Simplify usage of the pool functions.

* Small tweak.

* Attempt at using apply to simplify the convnet definition.

2023-08-29 19:12:16 +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

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

Simplify usage of the pool functions. (#662)

2023-08-29 19:12:16 +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