File size: 261 Bytes
cc2be96 |
1 2 3 4 5 6 7 8 9 10 11 12 |
# pixel resolution of the model
resolution: 512
# train 4 channel VAE latents instead of RGB at 1/8 resolution
vae_latents: true
# number of channels at each level of the UNet
channels: [128, 256, 384, 512]
# which levels have attention
attention: [0, 1, 2]
|