clement

### Describe the bug I am trying to finetune SDXL but the training script crashes when saving the model at a checkpoint. Training runs fine. ### Reproduction Here is my...