Changes to fix working for transformer-xl
Update arguments (`lm_labels` to `labels`)
Use labels as data; labels are shifted inside the model (https://github.com/bajajahsaas/transformers/blob/master/src/transformers/modeling_transfo_xl.py#L855)
Sorry, it's been a while, but this has since been fixed in #3716.