hi @Winton_Cape seems like this your first time in the community.It seems like you issue is same as in this thread InvalidArgumentError: required broadcastable shapes [Op:Mul]  8
You could try running and make sure to use just cpu if you are on google colab because there are issues with GPU currently as mentioned in that thread. And I don’t know if that’s been fixed.
!pip install tensorflow==2.5.0

if you really want to use gpu and are comfortable switching between nightly version (I would recommend this considering you are doing a tutorial part with not trivial dataset), you can just use this
!pip install tflite-model-maker-nightly