diff --git a/.github/workflows/test_sdxl.yml b/.github/workflows/test_sdxl.yml
index bf97cc39..b0e59b5d 100644
--- a/.github/workflows/test_sdxl.yml
+++ b/.github/workflows/test_sdxl.yml
@@ -9,7 +9,7 @@ jobs:
     strategy:
       matrix:
         version: [3.11]
-        os: [nodai-amdgpu-w7900-x86-64]
+        os: [nodai-amdgpu-mi250-x86-64]
 
     runs-on: ${{matrix.os}}
     steps:
diff --git a/.github/workflows/test_shark.yml b/.github/workflows/test_shark.yml
index 6f2e4b4e..45ada02d 100644
--- a/.github/workflows/test_shark.yml
+++ b/.github/workflows/test_shark.yml
@@ -49,6 +49,7 @@ jobs:
           cd $GITHUB_WORKSPACE/SHARK
           python${{ matrix.version }} -m venv shark.venv
           source shark.venv/bin/activate
+          pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu
           pip install -r requirements.txt --no-cache-dir
           pip install -e .
           python apps/shark_studio/tests/api_test.py