I downgraded the tensorflow version in Colab it runs, Colab usually installs version 2.17.0 I used this code to downgrade it #!pip install tensorflow==2.12 # install tensorflow 2.12 import tensorflow as tf print(tf.version)