Open-Sora/opensora/models/vae/__init__.py

mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-05-21 11:59:01 +02:00
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
format (#69) 2024-03-15 15:16:20 +01:00			`from .vae import VideoAutoencoderKL, VideoAutoencoderKLTemporalDecoder`
register module 2024-04-09 12:04:08 +02:00			`from .vae_3d import VAE_3D, VAE_3D_B`
			`from .vae_3d_v2 import VAE_3D_V2, VAE_MAGVIT_V2`
added vae3d training code 2024-03-28 08:12:20 +01:00