ViT Support For CUDA Versions >11.8

[Joeljensen28]

Hi, the work here is great, and I am really interested in using it for one of my projects. However, I am unable to use the ViT model to predict normals, as the hardware I am using is not compatible with CUDA <12. The torch version in the requirements for the ViT model is set to 2.0.1, which is not compatible with CUDA >11.8, making it impossible for me to use the model with my current hardware.

Will there ever be support for CUDA 12?

Thanks!