BitDance_B_4x/autoencoder/config.json

{
  "_class_name": "BitDanceImageNetAutoencoder",
  "_diffusers_version": "0.36.0",
  "ddconfig": {
    "double_z": false,
    "z_channels": 32,
    "in_channels": 3,
    "out_ch": 3,
    "ch": 256,
    "ch_mult": [
      1,
      1,
      2,
      2,
      4
    ],
    "num_res_blocks": 4
  },
  "source_checkpoint": "/data/projects/BitDance/models/shallowdream204/BitDance-ImageNet/ae_d16c32.pt",
  "num_codebooks": 4
}