trax
trax copied to clipboard
ReformerLM api changes in text-gen notebook
It seems like API changed a bit, this fixes notebook.