fix: tokenizer -> processing_class SFTTrainer API change

2026-06-30 20:38:31 -04:00
parent 9d5d4eebc7
commit 83ba987840
1 changed files with 1 additions and 1 deletions
--- a/training/scripts/train.py
+++ b/training/scripts/train.py
@@ -99,7 +99,7 @@ def train(config_path):
    
    trainer = SFTTrainer(
        model=model,
-        tokenizer=tokenizer,
+        processing_class=tokenizer,
        train_dataset=dataset["train"],
        eval_dataset=dataset["test"],
        args=training_args,