Update build.sh

Author: svekars

.jenkins/build.sh

@@ -24,9 +24,8 @@ sudo apt-get install -y pandoc
 # Install 2.5 to merge all 2.4 PRs - uncomment to install nightly binaries (update the version as needed).
 sudo pip uninstall -y torch torchvision torchaudio torchtext torchdata torchrl tensordict
 pip3 install torch==2.7.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu126
-pip install git+https://github.com/pytorch/tensordict
-pip install git+https://github.com/pytorch/torchrl
-
+pip install git+https://github.com/pytorch/tensordict.git@main --depth=1
+pip install git+https://github.com/pytorch/torchrl.git@main --depth=1
 #sudo pip uninstall -y fbgemm-gpu torchrec
 #sudo pip3 install fbgemm-gpu==1.1.0 torchrec==1.0.0 --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu126