from .vae import VideoAutoencoderKL, VideoAutoencoderKLTemporalDecoder
from .vae_3d import VAE_3D, VAE_3D_B
from .vae_3d_v2 import VAE_3D_V2, VAE_MAGVIT_V2