Update setup.py to limit llguidance version (#1080) #1004
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: Pull Request Gate | |
on: | |
push: | |
branches: [main] | |
pull_request: | |
branches: [main] | |
workflow_dispatch: | |
jobs: | |
# First Stage ======================================================================= | |
# Linting and basic CPU-based tests | |
bare-install: | |
strategy: | |
matrix: | |
os: [ubuntu-latest, windows-latest, macos-12] | |
python-version: ["3.9", "3.10", "3.11", "3.12"] | |
runs-on: ${{ matrix.os }} | |
steps: | |
- uses: actions/checkout@v4 | |
- name: Set up Python ${{ matrix.python-version }} | |
uses: actions/setup-python@v5 | |
with: | |
python-version: ${{ matrix.python-version }} | |
- name: Minimal install | |
run: | | |
pip install -e . | |
- name: Attempt import | |
run: | | |
python -c "import guidance" | |
basic-tests-linux-python-latest: | |
uses: ./.github/workflows/action_plain_basic_tests.yml | |
strategy: | |
fail-fast: false # Don't cancel all on first failure | |
matrix: | |
model: | |
- "transformers_gpt2_cpu" | |
- "transformers_phi2_cpu" | |
# - "transformers_mistral_7b_cpu" See Issue 713 | |
- "llamacpp_llama2_7b_cpu" | |
- "llamacpp_mistral_7b_cpu" | |
- "transformers_phi3_mini_4k_instruct_cpu" | |
- "llamacpp_phi3_mini_4k_instruct_cpu" | |
with: | |
os: Large_Linux | |
python-version: "3.12" | |
model: ${{ matrix.model }} | |
end-stage-1: | |
needs: | |
- bare-install | |
- basic-tests-linux-python-latest | |
name: End Stage 1 | |
runs-on: ubuntu-latest | |
steps: | |
- run: echo "End of Stage 1" | |
# Second stage ======================================================================= | |
# Remaining CPU-based tests on Linux, plus server tests and basic GPU testing | |
basic-tests-gpu-python-latest: | |
needs: end-stage-1 | |
uses: ./.github/workflows/action_gpu_basic_tests.yml | |
with: | |
os: gpu-runner | |
python-version: "3.12" | |
basic-tests-linux-python-other: | |
needs: end-stage-1 | |
strategy: | |
fail-fast: false # Don't cancel all on first failure | |
matrix: | |
python-version: ["3.9", "3.10", "3.11"] | |
model: | |
- "transformers_gpt2_cpu" | |
- "transformers_phi2_cpu" | |
# - "transformers_mistral_7b_cpu" See Issue 713 | |
- "llamacpp_llama2_7b_cpu" | |
- "llamacpp_mistral_7b_cpu" | |
- "transformers_phi3_mini_4k_instruct_cpu" | |
- "llamacpp_phi3_mini_4k_instruct_cpu" | |
uses: ./.github/workflows/action_plain_basic_tests.yml | |
with: | |
os: Large_Linux | |
python-version: ${{ matrix.python-version }} | |
model: ${{ matrix.model }} | |
server-tests: | |
needs: end-stage-1 | |
strategy: | |
fail-fast: false # Don't cancel all on first failure | |
matrix: | |
# Need to figure out what happened to 3.12 | |
python-version: ["3.9", "3.10", "3.11", "3.12"] | |
uses: ./.github/workflows/action_server_tests.yml | |
with: | |
os: ubuntu-latest | |
python-version: ${{ matrix.python-version }} | |
end-stage-2: | |
needs: | |
- basic-tests-linux-python-other | |
- basic-tests-gpu-python-latest | |
- server-tests | |
name: End Stage 2 | |
runs-on: ubuntu-latest | |
steps: | |
- run: echo "End of Stage 1" | |
# Third Stage ============================================================== | |
# Windows and MacOS, plus other GPU Linux tests | |
basic-tests-mac-x86: | |
needs: end-stage-2 | |
strategy: | |
fail-fast: false # Don't cancel all on first failure | |
matrix: | |
python-version: ["3.9", "3.10", "3.11", "3.12"] | |
model: | |
- "transformers_gpt2_cpu" | |
- "transformers_phi2_cpu" | |
# - "transformers_mistral_7b_cpu" See Issue 713 | |
- "llamacpp_llama2_7b_cpu" | |
- "llamacpp_mistral_7b_cpu" | |
# - "transformers_phi3_mini_4k_instruct_cpu" Gives trouble on MacOS | |
- "llamacpp_phi3_mini_4k_instruct_cpu" | |
uses: ./.github/workflows/action_plain_basic_tests.yml | |
with: | |
os: macos-12 | |
python-version: ${{ matrix.python-version }} | |
model: ${{ matrix.model }} | |
basic-tests-mac-arm: | |
needs: end-stage-2 | |
strategy: | |
fail-fast: false # Don't cancel all on first failure | |
matrix: | |
python-version: ["3.9", "3.10", "3.11", "3.12"] | |
model: | |
# - "transformers_gpt2_cpu" See #965 | |
# - "transformers_phi2_cpu" Seems to get stuck | |
# - "transformers_mistral_7b_cpu" See Issue 713 | |
# - "llamacpp_llama2_7b_cpu" Getting stuck with llama-cpp-python 0.2.77 | |
# - "llamacpp_mistral_7b_cpu" | |
# - "transformers_phi3_mini_4k_instruct_cpu" Gives trouble on MacOS | |
- "llamacpp_phi3_mini_4k_instruct_cpu" | |
uses: ./.github/workflows/action_plain_basic_tests.yml | |
with: | |
os: macos-latest | |
python-version: ${{ matrix.python-version }} | |
model: ${{ matrix.model }} | |
basic-tests-win: | |
needs: end-stage-2 | |
strategy: | |
fail-fast: false # Don't cancel all on first failure | |
matrix: | |
python-version: ["3.9", "3.10", "3.11", "3.12"] | |
model: | |
- "transformers_gpt2_cpu" | |
- "transformers_phi2_cpu" | |
# - "transformers_mistral_7b_cpu" See Issue 713 | |
- "llamacpp_llama2_7b_cpu" | |
- "llamacpp_mistral_7b_cpu" | |
- "transformers_phi3_mini_4k_instruct_cpu" | |
- "llamacpp_phi3_mini_4k_instruct_cpu" | |
uses: ./.github/workflows/action_plain_basic_tests.yml | |
with: | |
os: Large_Windows | |
python-version: ${{ matrix.python-version }} | |
model: ${{ matrix.model }} | |
basic-tests-gpu-python-others: | |
needs: end-stage-2 | |
strategy: | |
fail-fast: false # Don't cancel all on first failure | |
matrix: | |
python-version: ["3.9", "3.10", "3.11"] | |
uses: ./.github/workflows/action_gpu_basic_tests.yml | |
with: | |
os: gpu-runner | |
python-version: ${{ matrix.python-version }} | |