nnels (int, optional, defaults to 3) — Number of channels in the latent space.   sample_size (int, optional, defaults to 32) — Sample input size.   num_vq_embeddings (int, optional, defaults to 256) — Number of codebook vectors in the VQ-VAE.   norm_num_groups (int, optional, defaults to 32) — Number of groups for normalization layers.   vq_embed_dim (int, optional) — Hidden dim of codebook vectors in the VQ-VAE.   scaling_factor (float, option