diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index a4930de9..10f0fa3c 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -174,7 +174,7 @@ jobs:
     concurrency:
       group: ${{ github.workflow }}-${{ github.job }}-${{ needs.build-and-push.outputs.label }}-${{ github.head_ref || github.run_id }}
       cancel-in-progress: true
-    if: needs.build-and-push.outputs.runs_on == 'amd-gpu-tgi'
+    if: needs.build-and-push.outputs.runs_on != 'ubuntu-latest'
     container:
       image: ${{ needs.build-and-push.outputs.docker_image }}
       options: --shm-size "16gb" --ipc host -v ${{ needs.build-and-push.outputs.docker_volume }}:/data
@@ -191,7 +191,7 @@ jobs:
             pwd
             echo "ls:"
             ls
-            python integration-tests/clean_cache_and_download.py --token ${{ secrets.HF_TOKEN }}
+            python integration-tests/clean_cache_and_download.py --token ${{ secrets.HF_TOKEN }} --cache-dir /data
           fi
 
   integration_tests:
@@ -243,4 +243,8 @@ jobs:
           echo "SYSTEM:"
           echo $SYSTEM
 
+          export DOCKER_VOLUME=${{ needs.build-and-push.outputs.docker_volume }}
+          echo "DOCKER_VOLUME:"
+          echo $DOCKER_VOLUME
+
           pytest -s -vvvvv integration-tests
diff --git a/integration-tests/clean_cache_and_download.py b/integration-tests/clean_cache_and_download.py
index d1011240..c2a3960c 100644
--- a/integration-tests/clean_cache_and_download.py
+++ b/integration-tests/clean_cache_and_download.py
@@ -35,7 +35,7 @@ REQUIRED_MODELS = {
 }
 
 
-def cleanup_cache(token: str):
+def cleanup_cache(token: str, cache_dir: str):
     # Retrieve the size per model for all models used in the CI.
     size_per_model = {}
     extension_per_model = {}
@@ -74,7 +74,7 @@ def cleanup_cache(token: str):
     total_required_size = sum(size_per_model.values())
     print(f"Total required disk: {total_required_size:.2f} GB")
 
-    cached_dir = huggingface_hub.scan_cache_dir()
+    cached_dir = huggingface_hub.scan_cache_dir(cache_dir)
 
     cache_size_per_model = {}
     cached_required_size_per_model = {}
@@ -121,7 +121,7 @@ def cleanup_cache(token: str):
 
         print("Removing", largest_model_id)
         for sha in cached_shas_per_model[largest_model_id]:
-            huggingface_hub.scan_cache_dir().delete_revisions(sha).execute()
+            huggingface_hub.scan_cache_dir(cache_dir).delete_revisions(sha).execute()
 
         del cache_size_per_model[largest_model_id]
 
@@ -135,10 +135,11 @@ if __name__ == "__main__":
     parser.add_argument(
         "--token", help="Hugging Face Hub token.", required=True, type=str
     )
+    parser.add_argument("--cache-dir", help="Hub cache path.", required=True, type=str)
     args = parser.parse_args()
 
     start = time.time()
-    extension_per_model = cleanup_cache(args.token)
+    extension_per_model = cleanup_cache(args.token, args.cache_dir)
     end = time.time()
 
     print(f"Cache cleanup done in {end - start:.2f} s")
@@ -153,6 +154,7 @@ if __name__ == "__main__":
             revision=revision,
             token=args.token,
             allow_patterns=f"*{extension_per_model[model_id]}",
+            cache_dir=args.cache_dir,
         )
     end = time.time()