Ensure input tensor are on device. (#11874)
The feature extractor does not create tensors on the appropriate device, so we call `ensure_tensor_on_device` before feeding the processed inputs to the model.
Showing
Please register or sign in to comment