diff --git a/pyproject.toml b/pyproject.toml index 506e9a7552..c7368b276a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -36,9 +36,9 @@ huggingface = [ "transformers>=4.37.0", "datasets", "sentencepiece", - "accelerate", + "accelerate>=0.26.1", "trl>=0.7.9", - "peft", + "peft>=0.7.1", "tqdm", "psutil", ]