From f5351c18e15cd26fe628ed364eaa4ecd49874596 Mon Sep 17 00:00:00 2001 From: hiyouga Date: Thu, 7 Sep 2023 19:26:25 +0800 Subject: [PATCH] update requirements --- requirements.txt | 2 +- src/llmtuner/tuner/core/loader.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index 0c725b8f..ff52658d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,7 +2,7 @@ torch>=1.13.1 transformers>=4.29.1 datasets>=2.12.0 accelerate>=0.21.0 -peft>=0.4.0 +peft==0.4.0 trl>=0.7.1 scipy sentencepiece diff --git a/src/llmtuner/tuner/core/loader.py b/src/llmtuner/tuner/core/loader.py index 17d66d78..f0e4afab 100644 --- a/src/llmtuner/tuner/core/loader.py +++ b/src/llmtuner/tuner/core/loader.py @@ -39,7 +39,7 @@ logger = get_logger(__name__) check_min_version("4.29.1") require_version("datasets>=2.12.0", "To fix: pip install datasets>=2.12.0") require_version("accelerate>=0.21.0", "To fix: pip install accelerate>=0.21.0") -require_version("peft>=0.4.0", "To fix: pip install peft>=0.4.0") +require_version("peft==0.4.0", "To fix: pip install peft==0.4.0") require_version("trl>=0.7.1", "To fix: pip install trl>=0.7.1")