Error using Trainer with Colab notebook, anyone have a solution?

Perhaps numpy>=2 issue…?

in Colab:

!pip install numpy==1.26.4