From 38e63bfd28179fbf4b06adc0d127c358fd297e58 Mon Sep 17 00:00:00 2001 From: hiyouga Date: Sun, 4 Feb 2024 00:01:16 +0800 Subject: [PATCH] bump up transformers version --- requirements.txt | 2 +- src/llmtuner/model/loader.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index 0e9090e4..c74f4fa8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ torch>=1.13.1 -transformers>=4.36.2 +transformers>=4.37.2 datasets>=2.14.3 accelerate>=0.21.0 peft>=0.7.0 diff --git a/src/llmtuner/model/loader.py b/src/llmtuner/model/loader.py index b02a4560..ed86a1c0 100644 --- a/src/llmtuner/model/loader.py +++ b/src/llmtuner/model/loader.py @@ -21,7 +21,7 @@ if TYPE_CHECKING: logger = get_logger(__name__) -require_version("transformers>=4.36.2", "To fix: pip install transformers>=4.36.2") +require_version("transformers>=4.37.2", "To fix: pip install transformers>=4.37.2") require_version("datasets>=2.14.3", "To fix: pip install datasets>=2.14.3") require_version("accelerate>=0.21.0", "To fix: pip install accelerate>=0.21.0") require_version("peft>=0.7.0", "To fix: pip install peft>=0.7.0")