Fix memory issue with llama.cpp LLM pipeline, closes #824 #1196
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
# GitHub Actions build workflow | |
name: build | |
on: ["push", "pull_request"] | |
jobs: | |
build: | |
runs-on: ${{ matrix.os }} | |
strategy: | |
matrix: | |
os: [ubuntu-latest, macos-latest, windows-latest] | |
timeout-minutes: 60 | |
steps: | |
- name: Checkout code | |
uses: actions/checkout@v3 | |
- name: Install Python | |
uses: actions/setup-python@v4 | |
with: | |
python-version: 3.9 | |
- name: Install Java | |
uses: actions/setup-java@v3 | |
with: | |
distribution: "zulu" | |
java-version: 17 | |
- name: Install dependencies - Linux | |
run: sudo apt-get update && sudo apt-get install libportaudio2 libsndfile1 | |
if: matrix.os == 'ubuntu-latest' | |
- name: Install dependencies - macOS | |
run: | | |
echo "OMP_NUM_THREADS=1" >> $GITHUB_ENV | |
echo "PYTORCH_MPS_DISABLE=1" >> $GITHUB_ENV | |
echo "LLAMA_NO_METAL=1" >> $GITHUB_ENV | |
echo "TIKA_STARTUP_SLEEP=30" >> $GITHUB_ENV | |
echo "TIKA_STARTUP_MAX_RETRY=10" >> $GITHUB_ENV | |
brew install portaudio | |
if: matrix.os == 'macos-latest' | |
- name: Install dependencies - Windows | |
run: choco install wget | |
if: matrix.os == 'windows-latest' | |
- name: Build | |
run: | | |
pip install -U wheel | |
pip install .[all,dev] fasttext==0.9.2 pillow==10.4.0 | |
pip cache purge | |
python -c "import nltk; nltk.download(['punkt', 'punkt_tab', 'averaged_perceptron_tagger_eng'])" | |
python --version | |
make data coverage | |
- uses: pre-commit/[email protected] | |
if: matrix.os == 'ubuntu-latest' | |
- name: Test Coverage | |
run: coveralls --service=github | |
if: matrix.os == 'ubuntu-latest' | |
env: | |
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |