Skip to content

Commit

Permalink
rm peft from pypi package since it wont install (#420)
Browse files Browse the repository at this point in the history
  • Loading branch information
vchiley authored Jul 3, 2023
1 parent 4e6a878 commit 904a321
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 3 deletions.
8 changes: 5 additions & 3 deletions .github/workflows/release.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -32,11 +32,13 @@ jobs:
PYPI_PACKAGE_NAME="llm-foundry-test-$(date +%Y%m%d%H%M%S)"
fi
# Remove the xentropy-cuda-lib and triton-pre-mlir dependencies as PyPI does not support
# direct installs. The error message for importing FusedCrossEntropy/flash_attn_triton
# gives instructions on how to install if a user tries to use it without this dependency.
# Remove the peft, xentropy-cuda-lib and triton-pre-mlir dependencies as PyPI does not
# support direct installs. The error message for importing PEFT, FusedCrossEntropy,
# and flash_attn_triton gives instructions on how to install if a user tries to use it
# without this dependency.
sed '/xentropy-cuda-lib@git+https:\/\/github.com\/HazyResearch\/flash-attention.git@.*/d' -i setup.py
sed '/triton-pre-mlir@git+https:\/\/github.com\/vchiley\/triton.git@.*/d' -i setup.py
sed '/peft@git+https:\/\/github.com\/huggingface\/peft.git@.*/d' -i setup.py
python -m pip install --upgrade build twine
python -m build
Expand Down
1 change: 1 addition & 0 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,6 +88,7 @@

extra_deps['peft'] = [
'loralib==0.1.1', # lora core
# PyPI does not support direct dependencies, so we remove this line before uploading from PyPI
'peft @ git+https://github.com/huggingface/peft.git', # TODO: pin it down only after it stabilizes.
'bitsandbytes==0.39.1', # 8bit
'scipy>=1.10.0,<=1.11.0', # bitsandbytes dependency; TODO: eliminate when incorporated to bitsandbytes
Expand Down

0 comments on commit 904a321

Please sign in to comment.