Instructions to use anjali-mudgal/roberta_base-LoRA with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use anjali-mudgal/roberta_base-LoRA with PEFT:
from peft import PeftModel from transformers import AutoModelForSequenceClassification base_model = AutoModelForSequenceClassification.from_pretrained("FacebookAI/roberta-base") model = PeftModel.from_pretrained(base_model, "anjali-mudgal/roberta_base-LoRA") - Notebooks
- Google Colab
- Kaggle
metadata
language: en
license: mit
datasets:
- jigsaw-toxic-comment-classification-challenge
- kaggle/jigsaw-toxic-comment-classification-challenge
tags:
- text-classification
base_model: FacebookAI/roberta-base
library_name: peft
Toxicity Detection Model
train_runtime': 501.1294 r = 4 num_train_epochs=1 optim="adamw_torch_fused", lr_scheduler_type="cosine", metric_for_best_model="f1",
Framework versions
- PEFT 0.14.0