Constant VQVAE_CODEBOOK_SIZE 

pub const VQVAE_CODEBOOK_SIZE: u32 = 4096;