Open-Sora/opensora/models/vae/__init__.py
2024-04-09 18:04:08 +08:00

5 lines
157 B
Python

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