How to setup mt5-xxl server with a checkpoint?

I am trying to run this notebook related to co referencing( https://github.com/google-research/google-research/blob/master/coref_mt5/colabs/coreference_decoder.ipynb ).

Under requirements heading it says:
Please start a mt5 sever with the checkpoint Coref-mT5-XXL model.
For instance, via rpc-calls send and get results. Not included in this colab.

The checkpoint is stored in google cloud storage: https://console.cloud.google.com/storage/browser/gresearch/correference_seq2seq/checkpoint_1140000

I tried downloading the checkpoint to colab and use transformers to access it but gives me this error: OSError: Error no file named pytorch model.bin, tf model.h5, model,ckpt. index or flax_model.msgpack found In directory /content/v2/checkpoint_1140000

Can someone help me set up this model along with its checkpoint?