mindspore-lab / mindone

one for all, Optimal generator with No Exception
https://mindspore-lab.github.io/mindone/
Apache License 2.0
338 stars 63 forks source link

Add checkpoint deletion feature to SDXL #286

Closed HaoyangLee closed 6 months ago

HaoyangLee commented 6 months ago

The sdxl ckpt files are too large (~13GB each) to save in local server especially args.save_ckpt_interval is set to be a small value. We can keep the latest args.max_num_ckpt ckpt files and delete those old ones. Keep all ckpts when args.max_num_ckpt is 0.