UNet trainer for SDXL models. Modified from Aozora.
https://github.com/Hysocs/Aozora_SDXL_Training 

Might not be suitable for production use. I used it for this model: https://huggingface.co/geltz/saeri    
