stmasson commited on
Commit
0d48090
·
verified ·
1 Parent(s): 517aef6

Upload scripts/train_n8n_sft.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. scripts/train_n8n_sft.py +1 -1
scripts/train_n8n_sft.py CHANGED
@@ -233,7 +233,7 @@ training_args = SFTConfig(
233
  save_total_limit=3,
234
  eval_strategy="steps",
235
  eval_steps=500,
236
- max_seq_length=MAX_SEQ_LENGTH,
237
  packing=True,
238
  gradient_checkpointing=True,
239
  gradient_checkpointing_kwargs={"use_reentrant": False},
 
233
  save_total_limit=3,
234
  eval_strategy="steps",
235
  eval_steps=500,
236
+ max_length=MAX_SEQ_LENGTH, # renamed from max_seq_length in TRL 0.12+
237
  packing=True,
238
  gradient_checkpointing=True,
239
  gradient_checkpointing_kwargs={"use_reentrant": False},