Can't build keras model after Tensorflow update from 2.2 to 2.3

We had a similar problem once we updated our Lambda workstation. TensorFlow got updated to version 2.5 and we could no longer run any Tensorflow models. The command

pip install -U protobuf

fixed the problem though.