diff --git a/.github/workflows/test-check.yaml b/.github/workflows/test-check.yaml index 6ea07f3ac5e..efe53099874 100644 --- a/.github/workflows/test-check.yaml +++ b/.github/workflows/test-check.yaml @@ -197,3 +197,27 @@ jobs: run: pip3 install .[dev,torchvision,onnxruntime] torch==1.9.1 - name: "🔬 Running onnx tests" run: make test TARGETS=onnx + transformers-tests: + runs-on: ubuntu-22.04 + env: + SPARSEZOO_TEST_MODE: "true" + needs: test-setup + if: ${{needs.test-setup.outputs.pytorch == 1}} + steps: + - uses: actions/setup-python@v4 + with: + python-version: '3.11' + - 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,torch,transformers] + - name: "🔬 Running transformers tests" + run: make test TARGETS=transformers