p12475398
/
LLaMA-Factory-Mirror
forked from p04798526/LLaMA-Factory-Mirror
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

bump up transformers version

This commit is contained in:
hiyouga 2024-02-04 00:01:16 +08:00
parent 51df865734
commit 38e63bfd28
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
requirements.txt
View File

@ -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

2
src/llmtuner/model/loader.py
View File

@ -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")