python --version
python -m pip install --upgrade pip

pip install -r colab_env.txt
# Install t5x
git clone --branch=main https://github.com/google-research/t5x
python -m pip install ./t5x