# create data collator
data_collator = DataCollatorWithPadding(tokenizer=tokenizer)