/u/threevox on reddit made a colab for playing with the checkpoint:

https://colab.research.google.com/drive/1FhH3HlN8Ps_Pr9OR6Qc...