Robin Carpentier

## 🐛 Bug Using torch.set_default_device to set anything else than "cpu" leads to a ValueError when calling model.predict. ### To Reproduce I'm using the code sample provided on the huggingface...