[lvwarren]

Make this change in utils.py:

  def load_gpt2_params_from_tf_ckpt(tf_ckpt_path, hparams):
       [...]
        #name = name.removeprefix("model/")
        name = name[len('model/'):]

and you're cool example will run in Google Colab under Python 3.8 otherwise the 3.9 Jupyter patching is a headache.