pull image explicitly

huggingface · Feb 22, 2024 · 4f1ed35 · 4f1ed35
1 parent 321adc0
commit 4f1ed35
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/.github/workflows/test_cli_cpu_py_tgi.yaml b/.github/workflows/test_cli_cpu_py_tgi.yaml
@@ -28,5 +28,8 @@ jobs:
           pip install --upgrade pip
           pip install -e .[testing,py-tgi]
 
+      - name: Pull TGI docker image
+        run: docker pull ghcr.io/huggingface/text-generation-inference:latest
+
       - name: Run tests
         run: pytest -k "cli and cpu and py_tgi"