types.


block_out_channels (Tuple[int], optional, defaults to  —
obj:(64,)): Tuple of block output channels.


act_fn (str, optional, defaults to "silu") — The activation function 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 (in