From b9f42172dd4bffb2f9e83acff9608e763ee396d4 Mon Sep 17 00:00:00 2001 From: hiyouga Date: Thu, 9 Nov 2023 13:45:10 +0800 Subject: [PATCH] fix #1438 #1439 Former-commit-id: 33422e1feff706bb7cfb235b9e40bd2e124222be --- 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 67dcb6b1..1c1bd308 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ torch>=1.13.1 transformers>=4.31.0,<4.35.0 -datasets>=2.12.0 +datasets>=2.14.0 accelerate>=0.21.0 peft>=0.6.0 trl==0.7.2 diff --git a/src/llmtuner/tuner/core/loader.py b/src/llmtuner/tuner/core/loader.py index 80d2c658..663f60d9 100644 --- a/src/llmtuner/tuner/core/loader.py +++ b/src/llmtuner/tuner/core/loader.py @@ -39,7 +39,7 @@ logger = get_logger(__name__) require_version("transformers>=4.31.0,<4.35.0", "To fix: pip install \"transformers>=4.31.0,<4.35.0\"") -require_version("datasets>=2.12.0", "To fix: pip install datasets>=2.12.0") +require_version("datasets>=2.14.0", "To fix: pip install datasets>=2.14.0") require_version("accelerate>=0.21.0", "To fix: pip install accelerate>=0.21.0") require_version("peft>=0.6.0", "To fix: pip install peft>=0.6.0") require_version("trl==0.7.2", "To fix: pip install trl==0.7.2")