train_vae显存不够

@bubbliiiing 我使用下面的命令训练vae，我看到autoencoder_kl_32x32x4_cogvideox.yaml的帧数video_len=49，但是我用48G显存的L40s GPU最大帧数只能开到13，我的视频分辨率是1440*1080，请问你们训练为什么帧数可以开到49？是视频分辨率更低吗？烦请拨冗回答一下 torchrun --master_addr=127.0.0.1 --master_port=6006 --nnodes=1 --node_rank=0 --nproc_per_node=4 scripts/train_vae.py \ -t \ -b easyanimate/vae/configs/autoencoder/autoencoder_kl_32x32x4_cogvideox.yaml \ --scale_lr=False \ --precision=bf16 \ --max_steps=2000 \ --resume="True"