Disk offload fix
# What does this PR do?

This PR fixes the disk offloading for pretrained models (requires latest accelerate main branch) and adds a test. The test passes locally for GPT-2, GPT-J, OPT and T5 (the only model where it's activated right now).
_The documentation is not available anymore as the PR was closed or merged._