From 8ed03f5f45cc31cc72a85df57c42112c88da5bcd Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Sat, 27 Jan 2024 17:38:30 +1100 Subject: [PATCH] Update pyproject.toml --- pyproject.toml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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", ]