Open-Sora/opensora/models/vae/__init__.py
zhengzangw a2dc903863 update
2024-04-29 09:41:39 +00:00

3 lines
116 B
Python

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