ortex icon indicating copy to clipboard operation
ortex copied to clipboard

Add example for BERT sequence classification

Open victor23k opened this issue 1 year ago • 1 comments

Hello! :heart: the library, I'm using it right now to run inference on a BERT fine-tuned for sequence classification.

@mortont do you think it would be helpful to add an example on how to set up the pipeline for inference similar to the StableLM example? The main thing that I did not find documented anywhere (maybe I didn't looked hard enough) was getting the id2label mappings from the generated config.json. If you think it will add value, I'll prepare the PR.

victor23k avatar May 16 '24 16:05 victor23k

Thanks for the feedback! Yes a PR would definitely be welcome, anything you found undocumented along the way could help others get started.

mortont avatar May 16 '24 19:05 mortont