fix: pass full Processor as processing_class for VLM SFTTrainer
This commit is contained in:
parent
a4346b954e
commit
202b7cdfa7
1 changed files with 1 additions and 1 deletions
|
|
@ -792,7 +792,7 @@ class UnslothTrainer:
|
|||
trainer_kwargs = {
|
||||
"model": self.model,
|
||||
"train_dataset": dataset['dataset'],
|
||||
"processing_class": self.tokenizer.tokenizer,
|
||||
"processing_class": self.tokenizer,
|
||||
"data_collator": data_collator,
|
||||
"args": SFTConfig(**config_args),
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue