So, if you were trying to do inference with your model. i.e. output = m1(input). What exactly would your input variable be? If it’s just batch try removing the .text from batch.text and see if that fixes your issue.