Update common.libsonnet

GoogleCloudPlatform · Sep 30, 2024 · 4a4a17d · 4a4a17d
1 parent a266d0a
commit 4a4a17d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/dags/legacy_test/tests/pytorch/r2.5/common.libsonnet b/dags/legacy_test/tests/pytorch/r2.5/common.libsonnet
@@ -24,7 +24,7 @@ local volumes = import 'templates/volumes.libsonnet';
     tpuSettings+: {
       softwareVersion: 'tpu-ubuntu2204-base',
     },
-    imageTag: 'r2.5.0-rc1_3.10',
+    imageTag: 'r2.5.0-rc5_3.10',
   },
   PyTorchTest:: common.PyTorchTest + r2_5 {
     local config = self,
@@ -104,7 +104,7 @@ local volumes = import 'templates/volumes.libsonnet';
         pip install pillow
         git clone --depth=1 https://github.com/pytorch/pytorch.git
         cd pytorch
-        git clone -b v2.5.0-rc1 https://github.com/pytorch/xla.git
+        git clone -b v2.5.0-rc5 https://github.com/pytorch/xla.git
       |||,
     },
     podTemplate+:: {
@@ -145,7 +145,7 @@ local volumes = import 'templates/volumes.libsonnet';
         pip install https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.5.0rc1-cp310-cp310-linux_x86_64.whl
 
         mkdir -p pytorch/xla
-        git clone -b v2.5.0-rc1 https://github.com/pytorch/xla.git pytorch/xla
+        git clone -b v2.5.0-rc5 https://github.com/pytorch/xla.git pytorch/xla
 
         %s