kyy/llm_trainer
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update transformers to 4.36.2 to resolve bug when saving a checkpoint in the multi-node setting.

Browse Source
This commit is contained in:
ShaneTian authored and GitHub committed 2023-12-20 22:00:41 +08:00
1 parent 7910dbae92
commit 390f0caf7f
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
requirements.txt
+1 -1
View File
@@ -1,5 +1,5 @@
torch>=1.13.1
transformers>=4.36.1
transformers>=4.36.2
datasets>=2.14.3
accelerate>=0.21.0
peft>=0.7.0