System Info

• transformers version: 4.30.0.dev0
• Platform: Linux-5.10.147+-x86_64-with-glibc2.31
• Python version: 3.10.11
• Huggingface_hub version: 0.14.1
• Safetensors version: not installed
• PyTorch version (GPU?): 2.0.0+cu118 (False)
• Tensorflow version (GPU?): 2.12.0 (False)
• Flax version (CPU?/GPU?/TPU?): 0.6.9 (cpu)
• Jax version: 0.4.8
• JaxLib version: 0.4.7
• Using GPU in script?: YES
• Using distributed or parallel set-up in script?: NO

Who can help?

@ArthurZucker

Information

• The official example scripts
• My own modified scripts

Tasks

• An officially supported task in the examples folder (such as GLUE/SQuAD, ...)
• My own task or dataset (give details below)

Reproduction

Run this notebook: https://colab.research.google.com/drive/1TFI84P9W4VPhNLgEngxPN57RwzS0C4bG?usp=sharing

Expected behavior

Expected the model to train successfully. Instead it gives a tensor mismatch error.