Skip to content

Commit

Permalink
make peft installs a extra_dep
Browse files Browse the repository at this point in the history
  • Loading branch information
vchiley committed Jun 29, 2023
1 parent 5a0495c commit a084c86
Showing 1 changed file with 7 additions and 4 deletions.
11 changes: 7 additions & 4 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -62,10 +62,6 @@
'cmake>=3.25.0,<=3.26.3', # required for triton-pre-mlir below
# PyPI does not support direct dependencies, so we remove this line before uploading from PyPI
'triton-pre-mlir@git+https://github.com/vchiley/triton.git@triton_pre_mlir_sm90#subdirectory=python',
'loralib==0.1.1', # lora core
'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
]

extra_deps = {}
Expand All @@ -90,6 +86,13 @@
'xentropy-cuda-lib@git+https://github.com/HazyResearch/[email protected]#subdirectory=csrc/xentropy',
]

extra_deps['peft'] = [
'loralib==0.1.1', # lora core
'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
]

extra_deps['all'] = set(dep for deps in extra_deps.values() for dep in deps)

setup(
Expand Down

0 comments on commit a084c86

Please sign in to comment.