fix: remove max_seq_length from SFTTrainer (newer TRL API)
This commit is contained in:
@@ -103,7 +103,6 @@ def train(config_path):
|
|||||||
train_dataset=dataset["train"],
|
train_dataset=dataset["train"],
|
||||||
eval_dataset=dataset["test"],
|
eval_dataset=dataset["test"],
|
||||||
args=training_args,
|
args=training_args,
|
||||||
max_seq_length=config["train_params"]["max_seq_length"],
|
|
||||||
)
|
)
|
||||||
|
|
||||||
# Train
|
# Train
|
||||||
|
|||||||
Reference in New Issue
Block a user