Error when fine-tuning XLM-RoBERTa base on TF/Keras
Hello,

I am trying to finetune XLM-RoBERTa for text classification on tensorflow-keras. I am using the `TFXLMRobertaForSequenceClassification` class for training 

I am using google colab GPU for finetuning. Tensoflow version is 2.9.2

en_y_pred = model.predict(en_x_test_in, batch_size=128, verbose=1)
**InvalidArgumentError: indices[2,268] = 124030 is not in [0, 50265) [[node tf_roberta_for_sequence_classification_1/roberta/embeddings/Gather (defined at /usr/local/lib/python3.7/dist-packages/transformers/models/roberta/modeling_tf_roberta.py:149) ]] [Op:__inference_train_function_82886]

Errors may have originated from an input operation. Input Source operations connected to node tf_roberta_for_sequence_classification_1/roberta/embeddings/Gather: In[0] tf_roberta_for_sequence_classification_1/roberta/embeddings/Gather/resource:
In[1] IteratorGetNext (defined at /usr/local/lib/python3.7/dist-packages/keras/engine/training.py:866)**