I think the repo does not work anymore, I use Google Colab open the notebook, run the whole thing, but failed at XFormer module loading.