I've made a few improvements on how I train a lora, I might revisit this one.