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

4 lines
120 B
Python

from .vae import VideoAutoencoderKL, VideoAutoencoderKLTemporalDecoder
from .vae_3d_v2 import VAE_3D_V2, VAE_MAGVIT_V2