bump up transformers version

xin-li-67 · Feb 3, 2024 · 38e63bf · 38e63bf
1 parent 51df865
commit 38e63bf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -21,7 +21,7 @@
 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")