Skip to content

Commit

Permalink
📦 Update peft to make it work with llama for PT inferencing
Browse files Browse the repository at this point in the history
Signed-off-by: gkumbhat <[email protected]>
  • Loading branch information
gkumbhat committed Jan 29, 2024
1 parent c9f4bf4 commit 7cd5db9
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -28,11 +28,7 @@ dependencies = [
"torch>=2.0.1",
"tqdm>=4.65.0",
"transformers>=4.36.0",
# GK-AUG-25-2023 NOTE: mpt branch on Mayank's fork was merged to peft main on Aug 24 and it got deleted
# which broke caikit-nlp build. peft hasn't released newer version yet, so to get
# the build fix, we pulling peft from main branch commit. In future, we will pull PEFT from
# pypi
"peft@git+https://github.com/huggingface/peft.git@8c17d556a8fe9522e10d73d7bd3fad46a6ecae14"
"peft>=0.7.0,<0.8.0",
]

[tool.setuptools.packages.find]
Expand Down

0 comments on commit 7cd5db9

Please sign in to comment.