|
|
@@ -460,7 +460,7 @@ def train(
|
|
|
save_strategy=t_cfg["save_strategy"],
|
|
|
save_steps=t_cfg["save_steps"],
|
|
|
save_total_limit=t_cfg["save_total_limit"],
|
|
|
- max_seq_length=t_cfg["max_seq_length"],
|
|
|
+ max_length=t_cfg["max_seq_length"],
|
|
|
gradient_checkpointing=t_cfg["gradient_checkpointing"],
|
|
|
gradient_checkpointing_kwargs=t_cfg.get(
|
|
|
"gradient_checkpointing_kwargs", {"use_reentrant": False}
|