Hi Thanks for the response. In both seneraos I reset notebook to factory settings first.
OPTION 1

Set notebook settings in colab to NONE
Ran all code to the evaluate cell and then got error 'Incompatible shapes: [1,25] v.x [4] [op:Mul]

OPTION 2

Installed tensorflow==2.5.0 and then installed modelmaker nightly
When I tried to install required packages (the second notebook cell), it gave an error on the line from tflite_model_maker.config import ExportFormat
AttributeError: module ‘tensorflow.python.training.experimental.mixed_precision’ has no attribute ‘_register_wrapper_optimizer_cls’

As in the original issue, I remember that I got this working a couple of weeks ago, I am just trying to get back to that point.