search

huggingface / api-inference-community

Apache License 2.0
161 stars 61 forks source link

SetFit: Cast probabilities to float & increment version #411

Closed tomaarsen closed 6 months ago

tomaarsen commented 6 months ago

Hello!

Pull Request overview

Details

This PR should mostly speak for itself. The fix solves a bug that was causing issues when a torch SetFitHead is used.