Can't get attribute 'Corpus' on <module '__main__' from 'convert_transfo_xl_checkpoint_to_pytorch.py'>
I trained my data with the original transformer_xl  repo, but I use convert_transfo_xl_checkpoint_to_pytorch.py to transfer tf to pytorch,
error occurs：
AttributeError: Can't get attribute 'Corpus' on <module '__main__' from 'convert_transfo_xl_checkpoint_to_pytorch.py'>

to use my data, What code do I want to change?