Skip to content

Commit

Permalink
TST Install bitsandbytes for compile tests (#1796)
Browse files Browse the repository at this point in the history
Also, remove outdated comment.
  • Loading branch information
BenjaminBossan authored May 23, 2024
1 parent 3cf5359 commit b292256
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions .github/workflows/torch_compile_tests.yml
Original file line number Diff line number Diff line change
@@ -1,7 +1,5 @@
name: torch compile tests

# see peft/tests/__init__.py

on:
workflow_dispatch:
inputs:
Expand Down Expand Up @@ -33,6 +31,7 @@ jobs:
run: |
python -m pip install --upgrade pip
python -m pip install .[test]
python -m pip install bitsandbytes
if [ "${{ github.event.inputs.pytorch_nightly }}" = "true" ]; then
python -m pip install --upgrade --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu
fi
Expand Down

0 comments on commit b292256

Please sign in to comment.