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