cmedina/agenx-lora-training
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: remove max_seq_length from TrainingArguments (passed to SFTTrainer)

Browse Source
This commit is contained in:
Christian Medina
2026-06-30 20:17:04 -04:00
parent 7181acc580
commit 7f7467ef89
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
training/scripts/train.py
View File

@@ -74,7 +74,7 @@ def train(config_path):
}, },
) )
# Training arguments # Training arguments (max_seq_length removed - passed to SFTTrainer instead)
training_args = TrainingArguments( training_args = TrainingArguments(
output_dir=config["train_params"]["output_dir"], output_dir=config["train_params"]["output_dir"],
num_train_epochs=config["train_params"]["num_train_epochs"], num_train_epochs=config["train_params"]["num_train_epochs"],
@@ -84,7 +84,6 @@ def train(config_path):
lr_scheduler_type=config["train_params"]["lr_scheduler_type"], lr_scheduler_type=config["train_params"]["lr_scheduler_type"],
weight_decay=config["train_params"]["weight_decay"], weight_decay=config["train_params"]["weight_decay"],
warmup_ratio=config["train_params"]["warmup_ratio"], warmup_ratio=config["train_params"]["warmup_ratio"],
max_seq_length=config["train_params"]["max_seq_length"],
logging_steps=config["train_params"]["logging_steps"], logging_steps=config["train_params"]["logging_steps"],
save_steps=config["train_params"]["save_steps"], save_steps=config["train_params"]["save_steps"],
save_total_limit=config["train_params"]["save_total_limit"], save_total_limit=config["train_params"]["save_total_limit"],