Files
candle/candle-nn
Laurent Mazare 2746f2c4be DiffNeXt/unet (#859)
* DiffNeXt/unet

* Start adding the vae.

* VAE residual block.

* VAE forward pass.

* Add pixel shuffling.

* Actually use pixel shuffling.
2023-09-15 10:14:02 +01:00
..
2023-09-15 10:14:02 +01:00

candle-nn