Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
from transformers import AutoModelForDocumentQuestionAnswering
model = AutoModelForDocumentQuestionAnswering.from_pretrained(model_checkpoint)
In the [TrainingArguments] use output_dir to specify where to save your model, and configure hyperparameters as you see fit.