hey @vinurad13 what shape does test_encodings["input_ids"] have?
without seeing the details behind X_test my guess is that you need to reshape your inputs so that input_ids has shape (batch_size, max_seq_length)