Já pensou em usar a api da OpenAI? Com o GPT-3.5-turbo-16k, vc consegue colocar muito texto, dá pra treinar legal, + todo o conhecimento do GPT.