Hi @marton-avrios
Your first input format might work, and in prediction instead of feeding partial output you could ask the model to generate whole text. Passing partial output is also possible.
I had applied T5 to a somewhat similar problem, and it gave surprisingly good results.