File size: 258 Bytes
569d8e6 |
1 2 3 4 5 6 7 8 9 10 11 12 |
# 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]
|