From 6f7e07f5ce18041f9f5891669a9676ea664adf74 Mon Sep 17 00:00:00 2001 From: Alvaro Moran Date: Wed, 11 Sep 2024 08:50:30 +0000 Subject: [PATCH] WIP quick ci test --- .../workflows/test-pytorch-xla-tpu-tgi.yml | 20 +++++-------------- 1 file changed, 5 insertions(+), 15 deletions(-) diff --git a/.github/workflows/test-pytorch-xla-tpu-tgi.yml b/.github/workflows/test-pytorch-xla-tpu-tgi.yml index 4c681941..d88993ab 100644 --- a/.github/workflows/test-pytorch-xla-tpu-tgi.yml +++ b/.github/workflows/test-pytorch-xla-tpu-tgi.yml @@ -2,9 +2,7 @@ name: Optimum TPU / Test TGI on TPU on: push: - branches: [ main ] - paths: - - "text-generation-inference/**" + branches: [ quick-ci-test ] pull_request: branches: [ main ] paths: @@ -28,16 +26,8 @@ jobs: - name: Checkout uses: actions/checkout@v4 - - name: Build and test TGI server - run: | - HF_TOKEN=${{ secrets.HF_TOKEN_OPTIMUM_TPU_CI }} make tgi_test - - # Use a different step to test the Jetstream Pytorch version, to avoid conflicts with torch-xla[tpu] - - name: Install and test TGI server (Jetstream Pytorch) + - name: quick test run: | - pip install -U .[jetstream-pt] \ - -f https://storage.googleapis.com/jax-releases/jax_nightly_releases.html \ - -f https://storage.googleapis.com/jax-releases/jaxlib_nightly_releases.html \ - -f https://storage.googleapis.com/libtpu-releases/index.html - JETSTREAM_PT=1 HF_TOKEN=${{ secrets.HF_TOKEN_OPTIMUM_TPU_CI }} python -m \ - pytest -sv text-generation-inference/tests -k jetstream + python -m pip install transformers + HF_TOKEN=${{ secrets.HF_TOKEN_OPTIMUM_TPU_CI }} python -c "from transformers import AutoConfig; AutoConfig.from_pretrained('google/gemma-2b')" + exit 123