RLHF Pre-training Regularization Loss for Knowledge Retention - Coding problem | OneNoughtOne