[Tests] Fix test
# What does this PR do?

This PR fixes the failing SwinModelIntegrationTest.test_inference_image_classification_head.

It also makes sure the appropriate device is used in one of ViLT's integration tests, although this test still fails due to the non-deterministic behaviour of the model. Setting torch.manual_seed(2) didn't help (it passes locally for me, but not on another machine).

Also removes a print statement in ViTMAE's test file.
_The documentation is not available anymore as the PR was closed or merged._