[M1 Chip] Tensorflow can be imported but CNN doesn’t work

Hello, I have followed this website: https://github.com/jeffheaton/t81_558_deep_learning/blob/master/install/tensorflow-install-mac-metal-jul-2021.ipynb on how to install tensorflow on my macbook. However, my code stops when I try to run CNN and LSTM.

My cell will be marked as completed but I will still be at epoch 1 /3. Then the whole code will fail.

This is where I get the code from: https://keras.io/examples/nlp/text_classification_from_scratch/

PS: Code runs on Google Colab