[BART] test_dummy_inputs fails on GPU
```
RuntimeError: Expected object of device type cuda but got device type cpu for argument #3 'index' in call to _th_index_select

```
Easy fix, but putting here so I don't forget!