wip

.github/workflows/debug-dind.yml

@@ -1,7 +1,8 @@
 name: Debug DIND
 
 on:
-  push:
+#   push:
+  workflow_dispatch:
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}

.github/workflows/test-pytorch-xla-tpu-tgi-integration.yml

@@ -1,6 +1,7 @@
 name: Optimum TPU / Test TGI on TPU / Integration Tests
 
 on:
+  push:
   schedule:
     - cron: '0 4 * * *' # run at 4 AM UTC
     # This can be used to allow manually triggering nightlies from the web interface
@@ -32,9 +33,16 @@ jobs:
           sudo apt-get install -y python3 python3-pip
           sudo ln -s /usr/bin/python3 /usr/bin/python
 
+      # We cannot use make tpu-tgi here because we need to build the docker image with the network host option
       - name: Build TGI Docker Image
         run: |
-          make tpu-tgi
+          docker build --rm -f text-generation-inference/docker/Dockerfile \
+            --build-arg VERSION=$(VERSION) \
+            --build-arg TGI_VERSION=$(TGI_VERSION) \
+            --ulimit nofile=100000:100000 \
+            --network host \
+            -t huggingface/optimum-tpu:$(VERSION)-tgi .
+          docker tag huggingface/optimum-tpu:$(VERSION)-tgi huggingface/optimum-tpu:latest
 
       - name: Run integration tests
         run: |