diff --git a/setup.py b/setup.py index e72e4d36..3f86a8e7 100644 --- a/setup.py +++ b/setup.py @@ -19,7 +19,7 @@ 'datasets>=2.19.0', 'accelerate>=0.20.1', 'sentence_transformers', - 'peft', + 'peft==0.15.0', 'ir-datasets', 'sentencepiece', 'protobuf'