| # pixel resolution of the model | |
| resolution: 64 | |
| # train 4 channel VAE latents instead of RGB at 1/8 resolution | |
| vae_latents: false | |
| # number of channels at each level of the UNet | |
| channels: [128, 256, 384, 512] | |
| # which levels have attention | |
| attention: [1, 2] | |