[Text Generation][Doc] Point to KV Cache Injection #3295
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: Test Checks | |
on: | |
push: | |
branches: | |
- main | |
- 'release/*' | |
pull_request: | |
branches: | |
- main | |
- 'release/*' | |
# Allows you to run this workflow manually from the Actions tab | |
workflow_dispatch: | |
env: | |
SPARSEZOO_TEST_MODE: true | |
NM_DISABLE_ANALYTICS: "true" | |
jobs: | |
base-tests: | |
runs-on: ubuntu-20.04 | |
steps: | |
- uses: actions/checkout@v2 | |
- uses: actions/checkout@v2 | |
with: | |
repository: "neuralmagic/sparsezoo" | |
path: "sparsezoo" | |
ref: ${{needs.test-setup.outputs.branch}} | |
- name: "⚙️ Install sparsezoo dependencies" | |
run: pip3 install -U pip && pip3 install setuptools sparsezoo/ | |
- name: "Clean sparsezoo directory" | |
run: rm -r sparsezoo/ | |
- name: ⚙️ Install dependencies | |
run: pip3 install .[dev,server,image_classification,transformers] opencv-python | |
- name: Run base tests | |
run: make test | |
cli-smoke-tests: | |
runs-on: ubuntu-20.04 | |
steps: | |
- uses: actions/checkout@v2 | |
- uses: actions/checkout@v2 | |
with: | |
repository: "neuralmagic/sparsezoo" | |
path: "sparsezoo" | |
ref: ${{needs.test-setup.outputs.branch}} | |
- name: "⚙️ Install sparsezoo dependencies" | |
run: pip3 install -U pip && pip3 install setuptools sparsezoo/ | |
- name: "Clean sparsezoo directory" | |
run: rm -r sparsezoo/ | |
- name: ⚙️ Install dependencies | |
run: pip3 install .[dev,server,image_classification,transformers] | |
- name: Run CLI smoke tests | |
run: PYTEST_ARGS="-m smoke" make test TARGETS=cli,nobase | |
integrations-tests: | |
runs-on: ubuntu-20.04 | |
steps: | |
- uses: actions/checkout@v2 | |
- uses: actions/checkout@v2 | |
with: | |
repository: "neuralmagic/sparsezoo" | |
path: "sparsezoo" | |
ref: ${{needs.test-setup.outputs.branch}} | |
- name: "⚙️ Install sparsezoo dependencies" | |
run: pip3 install -U pip && pip3 install setuptools sparsezoo/ | |
- name: "Clean sparsezoo directory" | |
run: rm -r sparsezoo/ | |
- name: ⚙️ Install dependencies | |
run: pip3 install .[dev,server,image_classification,transformers,haystack] | |
- name: Run integrations tests | |
run: make test_integrations |