Hi ! Not 100% sure to know what’s going here, but can you check that you’re able to instantiate a torch tensor from a numpy array in your environment ? And that you didn’t pass any device= parameter to encoded_dataset.with_format() ?
Hi ! Not 100% sure to know what’s going here, but can you check that you’re able to instantiate a torch tensor from a numpy array in your environment ? And that you didn’t pass any device= parameter to encoded_dataset.with_format() ?