Video course · 5 chapters · 43 min · certificate
Build a Stable Diffusion VAE From Scratch using Pytorch
Build the variational autoencoder behind Stable Diffusion in PyTorch: VAE concepts, model blocks, encoder and decoder, training and plugging the trained VAE into a diffusion pipeline.
What you'll learn
- Explain how a VAE works
- Implement VAE building blocks in PyTorch
- Build the encoder and decoder
- Train a VAE
- Integrate a VAE with diffusion
Chapters
5 chapters · 43:18-
3:36
01Concepts Members
Understanding VAEs
Variational autoencoders.
-
13:29
02Blocks Members
Implementing model blocks
Residual and attention.
-
7:46
03Architecture Members
Building encoder and decoder
Assemble it.
-
10:12
04Training Members
Training the VAE
Reconstruct images.
-
8:15
05Diffusion Members
Integration with diffusion
Use the trained VAE.