I am working on a project to pre-train a custom transformer model I developed and then fine-tune it for