Open-Sora icon indicating copy to clipboard operation
Open-Sora copied to clipboard
verified publisher icon hpcaitech

Use VAE for video compress causes OOM

Open CacacaLalala opened this issue 1 year ago • 0 comments

I fisrt use VQVAE for video compress, the code runs fine. But Loss drops very slowly.

So I change AE to VAE, and I got the OOM error, even though I set batch_size and accumulation_steps to 1.

Has anyone encountered this problem too?

CacacaLalala avatar Mar 14 '24 03:03 CacacaLalala