Mask2Former on multi-gpu cuda

Hi!

Looking into documentation seems that mask2former transformer doesnt support cuda implementations. Am I correct?

Reference: https://huggingface.co/docs/transformers/main/en/model_doc/mask2former#transformers.Mask2FormerForUniversalSegmentation.forward.pixel_values (there is no torch.cuda.FloatTensor option)

Thanks