MaskFormer: fix device on test
# What does this PR do?
This PR fixes some failed tests in MaskFormer due to not setting `torch_device`
_The documentation is not available anymore as the PR was closed or merged._