from .discriminator import DISCRIMINATOR_3D from .vae import VideoAutoencoderKL, VideoAutoencoderKLTemporalDecoder from .vae_3d import VAE_Temporal