tion to use.


latent_channels (int, optional, defaults to 3) — Number of channels in the latent space.


sample_size (int, optional, defaults to 32) — TODO


num_vq_embeddings (int, optional, defaults to 256) — Number of codebook vectors in the VQ-VAE.


vq_embed_dim (int, optional) — Hidden dim of codebook vectors in the VQ-VAE.


scaling_factor (float, optional, defaults to 0.18215) —
The component-wise standard deviation of the trained late