FLUX.2-4B DMD2 训练需要多大显存？

[zengkaiya]

尝试用1*8 80G H100 运行最近更新的 https://github.com/modelscope/DiffSynth-Studio/blob/main/examples/flux2/model_training/special/dmd2/FLUX.2-klein-base-4B-DMD2.sh 代码，训练一个epoch就oom。想请教下大概需要多少计算资源呢？