diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index d415f369..6a543488 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -181,6 +181,16 @@ jobs:
       - name: Install
         run: |
           make install-integration-tests
+      - name: Setup tmate session
+        uses: mxschmitt/action-tmate@v3
+      - name: Run tests
+        run: |
+          export DOCKER_VOLUME=/mnt/cache
+          export DOCKER_IMAGE=${{ needs.build-and-push.outputs.docker_image }}
+          export DOCKER_DEVICES=${{ needs.build-and-push.outputs.docker_devices }}
+          export HF_TOKEN=${{ secrets.HF_TOKEN }}
+          echo $DOCKER_IMAGE
+          pytest -s -vv integration-tests ${PYTEST_FLAGS}
       - name: Run tests
         run: |
           export DOCKER_VOLUME=/mnt/cache