v0.6.0

v0.6.0 OpenVINO exporter, model cards, and various quality of life improvements 🔥

To bring in the new year, this release comes with many bug fixes and quality of life improvements around using SetFit models. It also provides:

• an OpenVINO exporter that you can optimise your models for inference with. Check out the notebooks for an example.
• a dedicated model card with metadata and usage instructions. See here for an example output from push_to_hub(): https://huggingface.co/lewtun/setfit-new-model-card

Bug fixes and improvements

• Always install the checked-out setfit by @tomaarsen in #235
• Add SetFitModel.to by @tomaarsen in #229)
• Add distillation trainer example by @lewtun in #202
• Prevent overriding the sample size in sample_dataset by @tomaarsen in #231
• add related work in readme by @Yongtae723 in #239
• Fix seed in trainer.py by @danielkorat in #243
• Always display test coverage; add tests by @tomaarsen in #240
• Add Tom to list of maintainers by @lewtun in #253
• Add proper model card by @lewtun in #252
• Added support of OpenVINO export by @AlexKoff88 in #214
• Add has_differentiable_head property to SetFitModel by @zachschillaci27 in #257
• Resolve numpy.ndarray type error with predict_proba by @jegork in #207
• Refactor model_head initialization in SetFitModel by @zachschillaci27 in #263
• Feature/deprecate binary cross entropy loss by @blakechi in #203
• Fix type hints by @Yongtae723 in #266
• pass auth token to sentence transformer by @ken-myers in #277
• Add multi-target support to SetFitHead by @Yongtae723 and @OskarLiew in #272
• Automatically create summary table after scripts/setfit/run_fewshot.py by @tomaarsen in #262
• Fix squared optimization steps bug by @twerkmeister in #280
• Fix squared optimization steps bug in distillation trainer by @tomaarsen in #284
• Dynamic features in datasets based on model input names by @AleksanderObuchowski in #288
• Resolve SentenceTransformer resetting devices after moving a SetFitModel by @tomaarsen in #283
• add run_zeroshot.py; add functionality to data.get_templated_dataset() (formerly add_templated_examples()) by @danielkorat in #292
• Exclude compatibility versions from dev setup by @tomaarsen in #286

Significant community contributions

The following contributors have made significant changes to the library over the last release:

• @tomaarsen
  • Always install the checked-out setfit (#235)
  • Add SetFitModel.to (#229) (#236)
  • Prevent overriding the sample size in sample_dataset (#231)
  • Always display test coverage; add tests (#240)
  • Automatically create summary table after scripts/setfit/run_fewshot.py (#262)
  • Fix squared optimization steps bug in distillation trainer (#284)
  • Resolve SentenceTransformer resetting devices after moving a SetFitModel (#283)
  • Reformat according to the newest black version
  • Remove doubled space in warning message
  • Exclude compatibility versions from dev setup (#286)
• @Yongtae723
  • add related work in readme (#239)
  • Fix type hints (#266)
  • Add multi-target support to SetFitHead (#272)
• @danielkorat
  • Fix seed in trainer.py (#243)
  • add run_zeroshot.py; add functionality to data.get_templated_dataset() (formerly add_templated_examples()) (#292)
• @AlexKoff88
  • Added support of OpenVINO export (#214)