fix: tokenizer -> processing_class SFTTrainer API change
This commit is contained in:
1 parent
9d5d4eebc7
commit
83ba987840
1 file changed
+1
-1
@@ -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,
|
||||
|
||||
Reference in new issue
Block a user