config.json

{
  "_commit_hash": null,
  "attn_dim_head": 32,
  "channels": 1,
  "codebook_size": 512,
  "fsq_levels": [
    7,
    5,
    5,
    5,
    5
  ],
  "image_size": 256,
  "init_dim": 64,
  "layers": [
    "residual",
    "residual",
    "compress_space",
    "residual",
    "residual",
    "compress_space",
    "residual",
    "residual",
    "compress_space",
    "residual",
    "residual",
    "compress_space",
    "residual",
    "residual"
  ],
  "model_type": "pdetokenizer",
  "num_codebooks": 1,
  "num_groups": 8,
  "pad_mode": "circular",
  "perceptual_loss_weight": 0,
  "quantization_type": "vq",
  "temporal_compression": false,
  "transformers_version": null,
  "use_batch_norm": false,
  "use_gan": false,
  "use_revin": true,
  "use_style": false
}