Use multiple workers for DataLoader at prediction step for Trainer
# What does this PR do?

Fixes #17749 by adding a parameter in the DataLoader init call of the test data_loader, so that we can use multiple workers for data preparation during prediction time.

@sgugger, I tag you because this is a Trainer related PR
_The documentation is not available anymore as the PR was closed or merged._