SFTConfig accept max_length instead of max_seq_length
Browse files
bonus-unit1/bonus-unit1.ipynb
CHANGED
|
@@ -742,7 +742,7 @@
|
|
| 742 |
" gradient_checkpointing=True,\n",
|
| 743 |
" gradient_checkpointing_kwargs={\"use_reentrant\": False},\n",
|
| 744 |
" packing=True,\n",
|
| 745 |
-
"
|
| 746 |
")"
|
| 747 |
]
|
| 748 |
},
|
|
|
|
| 742 |
" gradient_checkpointing=True,\n",
|
| 743 |
" gradient_checkpointing_kwargs={\"use_reentrant\": False},\n",
|
| 744 |
" packing=True,\n",
|
| 745 |
+
" max_length=max_seq_length,\n",
|
| 746 |
")"
|
| 747 |
]
|
| 748 |
},
|