I think the tokenized texts are not of the same length as indicated by this warning message.
If you adjust the input length to be the same at each batch, I think the error will go away.