Fix CI: test_inference_for_pretraining in ViTMAEModelTest
# What does this PR do?

Fix CI (scheduled): `test_inference_for_pretraining` in `ViTMAEModelTest`: incorrect device

https://github.com/huggingface/transformers/runs/5809883645?check_suite_focus=true
_The documentation is not available anymore as the PR was closed or merged._