CI Fixes for test_models and SHARK testing

.github/workflows/test_models.yml

@@ -49,6 +49,11 @@ jobs:
           pip install --no-compile --pre --upgrade -r requirements.txt
           pip install --no-compile --pre -e .[testing]
           pip install --no-compile --pre --upgrade -e ${{ github.workspace }}/SHARK-Turbine/models -r ${{ github.workspace }}/SHARK-Turbine/models/requirements.txt
+          pip install \
+            --find-links https://iree.dev/pip-release-links.html \
+            --upgrade \
+            iree-compiler \
+            iree-runtime
       - name: Show current free memory
         run: |
           free -mh
@@ -85,6 +90,11 @@ jobs:
           pip install --no-compile --pre --upgrade -r requirements.txt
           pip install --no-compile --pre -e .[testing]
           pip install --no-compile --pre --upgrade -e ${{ github.workspace }}/SHARK-Turbine/models -r ${{ github.workspace }}/SHARK-Turbine/models/requirements.txt
+          pip install \
+            --find-links https://iree.dev/pip-release-links.html \
+            --upgrade \
+            iree-compiler \
+            iree-runtime
       - name: Show current free memory
         run: |
           free -mh

.github/workflows/test_shark.yml

@@ -36,9 +36,9 @@ jobs:
       - name: "Checkout SHARK"
         uses: actions/checkout@v4
         with:
-          repository: "nod-ai/SHARK.git"
+          repository: "nod-ai/SHARK"
           path: SHARK
-          ref: "iree-turbine-switch"
+          ref: "diffusers-version"
 
       - name: "Install SHARK"
         run: |