The OOM error happens in the example Notebook I referenced, which I found in https://huggingface.co/transformers/training.html#additional-resources 13. In my case, the error I am getting is another one: TypeError: new(): invalid data type 'str', which I don’t know what does it mean.
Edit: I removed the max_length argument and the error is still happening.