Fix vision import that accidentally we used cloud_tpu ops. The vision/beta models were broken.
https://github.com/tensorflow/models/commit/46bf0651e0f1b70c6a46b7a545b8e730b50d65b1
Updated pip package: https://pypi.org/project/tf-models-official/2.6.1/