Open-Sora/opensora/models/vae/__init__.py
2024-04-29 17:02:22 +00:00

4 lines
148 B
Python

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