Modelpublic

hpcai-tech/vqvae

sourceHugging Facemitupdated 3y agoView on Hugging Face

6likes29downloads

config.json22 linesDownload Raw Back to root

1{2  "architectures": [3    "VQVAE"4  ],5  "auto_map": {6    "AutoConfig": "configuration_vqvae.VQVAEConfig",7    "AutoModel": "modeling_vqvae.VQVAE"8  },9  "downsample": [10    2,11    4,12    413  ],14  "embedding_dim": 256,15  "model_type": "VQVAE",16  "n_codes": 2048,17  "n_hiddens": 240,18  "n_res_layers": 4,19  "torch_dtype": "float32",20  "transformers_version": "4.37.2"21}22