[Trainer] Allow passing image processor (#29896)
* Add image processor to trainer * Replace tokenizer=image_processor everywhere
This commit is contained in:
@@ -369,7 +369,7 @@ def main():
|
||||
args=training_args,
|
||||
train_dataset=ds["train"] if training_args.do_train else None,
|
||||
eval_dataset=ds["validation"] if training_args.do_eval else None,
|
||||
tokenizer=image_processor,
|
||||
image_processor=image_processor,
|
||||
data_collator=collate_fn,
|
||||
)
|
||||
|
||||
|
||||
@@ -458,7 +458,7 @@ def main():
|
||||
args=training_args,
|
||||
train_dataset=ds["train"] if training_args.do_train else None,
|
||||
eval_dataset=ds["validation"] if training_args.do_eval else None,
|
||||
tokenizer=image_processor,
|
||||
image_processor=image_processor,
|
||||
data_collator=collate_fn,
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user