feat: Fine Tune (Part 10) - Updating fine tuning py

[ishaansehgal99]

Reason for Change: Updating the fine tuning program. Include fine tuning and inference in one image.

todo: Add RLHF - basic done Ensure valid file paths interacted with ("/mnt") dirs - done Ensure compatibility with different datasets - basic done Test save strategies - todo in next pr