[Trainer] Undo #29896 (#30129)

* Undo

* Use tokenizer

* Undo data collator
This commit is contained in:
NielsRogge
2024-04-09 12:55:42 +02:00
committed by GitHub
parent ba1b24e07b
commit e9c23fa056
20 changed files with 24 additions and 41 deletions

View File

@@ -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,
image_processor=image_processor,
tokenizer=image_processor,
data_collator=collate_fn,
)

View File

@@ -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,
image_processor=image_processor,
tokenizer=image_processor,
data_collator=collate_fn,
)