Open-Sora/opensora/models/vae/__init__.py
2024-03-28 15:12:20 +08:00

4 lines
99 B
Python

from .vae import VideoAutoencoderKL, VideoAutoencoderKLTemporalDecoder
from .vae_3d import VAE_3D