Source code attached with training optimization. Trained on 3 epochs without PEFT (took about 2 hours).
