IndexError: Invalid key: 16 is out of bounds for size 0

Just add remove_unused_columns=False to TrainingArguments

5 Likes