ValueError: Multioutput target data is not supported with label binarization

After running the trainer for >5 hours, I get ValueError: Multioutput target data is not supported with label binarization

My train_dataset and eval_dataset have one text column, one labels column (binary) and one column for each label. So it's the same as in the [text-classification_multilabel.ipynb](https://github.com/huggingface/setfit/blob/main/notebooks/text-classification_multilabel.ipynb) example.

Any idea what could be going on? Thanks