from xvideo.models.vae.vae import XVAEChunkCausal