config.yml

params:
    image_size: 64
    model_channels: 704
    num_res_blocks: 3
    channel_mult: "1,2,3,4"
    attention_resolutions: "32,16,8"
    num_heads: 1
    num_head_channels: 64
    num_heads_upsample: -1
    use_scale_shift_norm: true
    dropout: 0.0
    resblock_updown: true
    encoder_dim: 4096
    encoder_channels: 2816
    in_channels: 3
    out_channels: 6
    activation: "gelu"
    att_pool_heads: 64
    disable_self_attentions: false
    precision: "16"