mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-19 22:02:06 +00:00
491ed9e11d
* Patch rust release. * Trying to remove the rust-toolchain hardcoded in action. * Upgrade rust toolchain. * Put back the toolchain ? * Fix neuron dockerfile. * Move to the proper version of Rust. * 1.85 since the GH action doesn't respect the override. * Typo. * Fixing the github action. * Fixing docker llamacpp. * Fixing the github action. * Update clippy.
67 lines
1.8 KiB
YAML
67 lines
1.8 KiB
YAML
name: Server Tests
|
|
|
|
on:
|
|
pull_request:
|
|
paths:
|
|
- ".github/workflows/tests.yaml"
|
|
- "server/**"
|
|
- "proto/**"
|
|
- "router/**"
|
|
- "launcher/**"
|
|
- "Cargo.lock"
|
|
- "rust-toolchain.toml"
|
|
|
|
concurrency:
|
|
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
|
|
cancel-in-progress: true
|
|
|
|
jobs:
|
|
run_tests:
|
|
runs-on:
|
|
group: aws-highmemory-32-plus-priv
|
|
steps:
|
|
- uses: actions/checkout@v4
|
|
- name: Set up Python
|
|
uses: actions/setup-python@v4
|
|
id: python
|
|
with:
|
|
python-version: 3.11
|
|
- uses: dtolnay/rust-toolchain@1.85.0
|
|
with:
|
|
components: rustfmt, clippy
|
|
- name: Install Protoc
|
|
uses: arduino/setup-protoc@v1
|
|
- name: Clean unused files
|
|
run: |
|
|
sudo rm -rf /usr/local/lib/android # will release about 10 GB if you don't need Android
|
|
sudo rm -rf /usr/share/dotnet # will release about 20GB if you don't need .NET
|
|
- name: Install
|
|
run: |
|
|
sudo apt update
|
|
sudo apt install python3.11-dev -y
|
|
pip install -U pip uv
|
|
uv venv
|
|
source ./.venv/bin/activate
|
|
make install-cpu
|
|
- name: Download locked kernels
|
|
run: |
|
|
source ./.venv/bin/activate
|
|
hf-kernels download server
|
|
- name: Run server tests
|
|
run: |
|
|
source ./.venv/bin/activate
|
|
uv pip install pytest
|
|
export HF_TOKEN=${{ secrets.HF_TOKEN }}
|
|
pytest -s -vv server/tests
|
|
- name: Pre-commit checks
|
|
run: |
|
|
pip install pre-commit
|
|
pre-commit install
|
|
pre-commit run --all-files
|
|
- name: Run Rust tests
|
|
run: |
|
|
cargo test
|
|
- name: Run Rust tests with google feature
|
|
run: |
|
|
cargo test --features google
|