From 4a4a17df089569f077d4b3aae0370c1d84fd6625 Mon Sep 17 00:00:00 2001
From: Manfei <41607353+ManfeiBai@users.noreply.github.com>
Date: Mon, 30 Sep 2024 00:06:34 -0700
Subject: [PATCH] Update common.libsonnet

---
 dags/legacy_test/tests/pytorch/r2.5/common.libsonnet | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/dags/legacy_test/tests/pytorch/r2.5/common.libsonnet b/dags/legacy_test/tests/pytorch/r2.5/common.libsonnet
index 353fc8a6..a3c9caf6 100644
--- 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