config.json

{
  "architectures": [
    "VQVAE"
  ],
  "auto_map": {
    "AutoConfig": "configuration_vqvae.VQVAEConfig",
    "AutoModel": "modeling_vqvae.VQVAE"
  },
  "downsample": [
    2,
    4,
    4
  ],
  "embedding_dim": 256,
  "model_type": "VQVAE",
  "n_codes": 2048,
  "n_hiddens": 240,
  "n_res_layers": 4,
  "torch_dtype": "float32",
  "transformers_version": "4.37.2"
}