WIP test #1
Workflow file for this run
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Optimum TPU / Test TGI on TPU | |
| on: | |
| push: | |
| branches: [ main, enable-ci-cd ] | |
| paths: | |
| - "text-generation-inference/**" | |
| - ".github/**" | |
| 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 |