Olá Kezia, tente utilizar as versões abaixo:

!pip install sentencepiece
!pip install tf-models-nightly==2.4.0.dev20210501
!pip install tf-nightly==2.6.0.dev20210501

Fabio