Test LLM packaging

onnx · Aug 28, 2024 · 8773d74 · 8773d74
1 parent 90d288b
commit 8773d74
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/.github/workflows/publish-to-test-pypi.yml b/.github/workflows/publish-to-test-pypi.yml
@@ -33,6 +33,9 @@ jobs:
           pip install dist/*.whl
           models=$(turnkey models-location --quiet)
           turnkey -i $models/selftest/linear.py discover export-pytorch
+          # Test LLMs as well
+          pip install "dist/*.whl[llm]"
+          lemonade -i facebook/opt-125m huggingface-load llm-prompt -p "Hello, my thoughts are"
       - name: Publish distribution package to PyPI
         if: startsWith(github.ref, 'refs/tags/v')
         uses: pypa/gh-action-pypi-publish@release/v1