bugbug icon indicating copy to clipboard operation
bugbug copied to clipboard

Add support for transformer models

Open marco-c opened this issue 2 years ago • 1 comments

Add a simple example model for detecting defects, both with finetuning and with using embeddings to train a xgboost model

Fixes #39

marco-c avatar Oct 27 '23 12:10 marco-c

It would also be nice to support LoRa (or at least add an example on how to use it). https://huggingface.co/blog/Lora-for-sequence-classification-with-Roberta-Llama-Mistral

marco-c avatar Nov 08 '23 23:11 marco-c