This error could be raised if the checkpoint file is corrupt and thus the reading fails. Could you check the file size of the stored checkpoint and the one used in Colab? Also, try to save a new checkpoint and load it in your other setup.