kontext lora finetuning OOM

[wanglaofei]

在h20上，微调kontext lora出现oom，有方法可以解决吗

[mi804]

经测试，单卡单图编辑 --max_pixels 1048576的设置下，kontext lora显存占用约40G，如果您出现了OOM，可以检查一下是否开启了 --use_gradient_checkpointing。此外，训练数据集里有多卡编辑且图像过多，也会提高显存峰值。