RuntimeError: Expected object of scalar type Byte but got scalar type Bool for argument #2 'other'

If I run the code with default arguments (and use data.txt from the repository) I get the following message:

```
Traceback (most recent call last):
File "C:/Users/matej/git/xlnet-Pytorch/main.py", line 89, in
num_predict=args.num_predict)
File "C:\Users\matej\git\xlnet-Pytorch\data_utils.py", line 345, in make_permute
reuse_len)
File "C:\Users\matej\git\xlnet-Pytorch\data_utils.py", line 292, in _local_perm
non_mask_tokens = (~is_masked) & non_func_tokens
RuntimeError: Expected object of scalar type Byte but got scalar type Bool for argument #2 'other'
```
I use Python 3.6.9 and PyTorch 1.2.0.