{
  "_class_name": "VQVAE",
  "_diffusers_version": "0.17.0",
  "_name_or_path": "/tank/ml/muse/muse-maskgit-pytorch/models/vit-s-vqgan-f4",
  "beta": 0.25,
  "dec": {
    "depth": 8,
    "dim": 512,
    "dim_head": 64,
    "dropout": 0,
    "image_size": 256,
    "mlp_dim": 2048,
    "num_head": 8,
    "out_channels": 3,
    "patch_size": 8
  },
  "embed_dim": 32,
  "enc": {
    "depth": 8,
    "dim": 512,
    "dim_head": 64,
    "dropout": 0,
    "image_size": 256,
    "in_channels": 3,
    "mlp_dim": 2048,
    "num_head": 8,
    "patch_size": 8
  },
  "n_embed": 8192
}