Updating CI torch version.

Author: githubsgi

distributed/tensor_parallelism/requirements.txt

@@ -3,4 +3,6 @@
 --pre
 --extra-index-url https://download.pytorch.org/whl/nightly/cu118
 --extra-index-url https://download.pytorch.org/whl/nightly/cu121
-torch >= 2.3.0.dev0; sys_platform == "linux"
+--extra-index-url https://download.pytorch.org/whl/nightly/cu126
+--extra-index-url https://download.pytorch.org/whl/nightly/cu128
+torch >= 2.7.1; sys_platform == "linux"