Resolve crash with predict_proba & multi-output

[tomaarsen]

Hello!

Pull Request overview

• Resolve crash with predict_proba & multi-output
• Tests

Details

If multi_output is used, then predict_proba will return a list of arrays/tensors. This isn't taken into consideration, so now we first stack that into a full array/tensor. This would happen e.g. with

model = SetFitModel.from_pretrained(
    "BAAI/bge-small-en-v1.5", multi_target_strategy="multi-output"
)

• Tom Aarsen

[HuggingFaceDocBuilderDev]

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.