Updated Torch and related dependency version. We are now at pytorch and lightning >= 2.0. Also support CUDA 11.7, 11.8, and 12.1.
Note
Export script is currently broken on torch 2.0 and 2.0.1. This is because onnx export doesn't work for multihead attention in this version. I've included the nightly torch release for 2.1 to make the export work. Alternatively users can downgrade to 1.13. Will create a new sub-release when torch 2.1 releases officially.