Skip to content

Commit

Permalink
feat(CI/CD): added workflow to test TGI
Browse files Browse the repository at this point in the history
  • Loading branch information
tengomucho committed Feb 27, 2024
1 parent 28eef0e commit 4fedc16
Showing 1 changed file with 33 additions and 0 deletions.
33 changes: 33 additions & 0 deletions .github/workflows/test_tpu_tgi.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
name: Optimum TPU / Test TGI on TPU

on:
push:
branches: [ main ]
paths:
- "text-generation-inference/**"
pull_request:
branches: [ main ]
paths:
- "text-generation-inference/**"

concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
cancel-in-progress: true

jobs:
do-the-job:
name: Run TGI tests
runs-on: [optimum-tpu] # run the configured Google TPU
steps:
- name: Checkout
uses: actions/checkout@v2
- name: Create venv and install Pytorch/XLA
run: |
python3 -m venv optimum-tpu-pytorch
source optimum-tpu-pytorch/bin/activate
python -m pip install -U pip
python -m pip install "torch~=2.2.0" "torch_xla[tpu]~=2.2.0" -f https://storage.googleapis.com/libtpu-releases/index.html
- name: Run TGI server python tests
run: |
source optimum-tpu-pytorch/bin/activate
make tgi_test

0 comments on commit 4fedc16

Please sign in to comment.