From 4dc825abe05c9afb9f66006fa6301c0c497fe2b7 Mon Sep 17 00:00:00 2001
From: Gerson Kroiz <49447208+gkroiz@users.noreply.github.com>
Date: Mon, 28 Aug 2023 15:58:21 -0700
Subject: [PATCH] fix sd-model nightly pt/xla import (#976)

---
 tests/pytorch/nightly/sd-model.libsonnet | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/pytorch/nightly/sd-model.libsonnet b/tests/pytorch/nightly/sd-model.libsonnet
index 78884ca8b..8d3c64c3b 100644
--- a/tests/pytorch/nightly/sd-model.libsonnet
+++ b/tests/pytorch/nightly/sd-model.libsonnet
@@ -64,8 +64,8 @@ local utils = import 'templates/utils.libsonnet';
         # taming-transformers and CLIP override existing torch and torchvision so we need to reinstall
         pip uninstall -y torch torchvision
         pip install --user \
-          https://storage.googleapis.com/pytorch-xla-releases/wheels/xrt/tpuvm/torch-nightly-cp310-cp310-linux_x86_64.whl \
-          'torch_xla[tpuvm] @ https://storage.googleapis.com/pytorch-xla-releases/wheels/xrt/tpuvm/torch_xla-nightly-cp310-cp310-linux_x86_64.whl'
+          https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch-nightly-cp310-cp310-linux_x86_64.whl \
+          'torch_xla[tpuvm] @ https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-nightly-cp310-cp310-linux_x86_64.whl'
         pip3 install --user --pre --no-deps torchvision --extra-index-url https://download.pytorch.org/whl/nightly/cpu
 
         # Setup data