Open-Sora VAE-v1.2 Weights

This repository stores the weights of the VAE released by the Open-Sora team. You can visit our project at:

• GitHub
• Gallery
• Gradio Demo

The weights are released together with Open-Sora v1.2.

We recommend you to use this weights in the Open-Sora codebase. If you want to use VAE in your own project, you may use the following sample code.

1. Install opensora

pip install git+https://github.com/hpcaitech/Open-Sora.git

2. Use STDiT3 in your own code

from opensora.models.vae.vae import VideoAutoencoderPipeline
vae = VideoAutoencoderPipeline.from_pretrained("hpcai-tech/OpenSora-VAE-v1.2")