@nielsr
yes, I’ve just looked at your changes However, I still have the same problems with model.generate () in relation to the validation step and the possibility of exporting the model to ONNX
PS: i have removed the decoder_input_ids after your push today but this notebook is not up to date