423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-03 04:02:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

bump up transformers version

Browse Source 
Former-commit-id: 38e63bfd28179fbf4b06adc0d127c358fd297e58
This commit is contained in:
hiyouga 2024-02-04 00:01:16 +08:00
parent 961368a8c4
commit b1064d2f9b
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 torch>=1.13.1
transformers>=4.36.2 transformers>=4.37.2
datasets>=2.14.3 datasets>=2.14.3
accelerate>=0.21.0 accelerate>=0.21.0
peft>=0.7.0 peft>=0.7.0

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

@ -21,7 +21,7 @@ if TYPE_CHECKING:
logger = get_logger(__name__) 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("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("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") require_version("peft>=0.7.0", "To fix: pip install peft>=0.7.0")