Грамотное дообучение модели ЯндексGPT

Есть датасет для Яндекс в нужном виде (но нужно почистить дополнительно и грамотно сделать. 2500 примеров. А также код к нему. Нужно переписать грамотно, сделать по новой файнтюн модели или же сделать рефакторинг.