iadb / cats64.yaml
pharmapsychotic's picture
pretrained cat models
cc2be96
# 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]