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(
|
trainer = SFTTrainer(
|
||||||
model=model,
|
model=model,
|
||||||
tokenizer=tokenizer,
|
processing_class=tokenizer,
|
||||||
train_dataset=dataset["train"],
|
train_dataset=dataset["train"],
|
||||||
eval_dataset=dataset["test"],
|
eval_dataset=dataset["test"],
|
||||||
args=training_args,
|
args=training_args,
|
||||||
|
|||||||
Reference in new issue
Block a user