Fix MP and CPU offload tests for Funnel and GPT-Neo
# What does this PR do?

This PR should fix the last failing GPU/multi-GPU tests.
_The documentation is not available anymore as the PR was closed or merged._