Perhaps your features (`output` in this case) have excessive nesting (inputs type `list` where type `int` is expected)

🤗Transformers
10

This smells like an unresolved issue…
OK, let’s work around it. If we process the dataset in advance, SFTTrainer won’t complain.