Hi @EdwinRijgersberg ! Thanks !
Since the last write-up, I adapted the code to use the latest HF and also Pytorch Lightning code.
You can check it out here:
  
      
      colab.research.google.com
  
  
    

Google Colaboratory 77