google/jigsaw_unintended_bias
Updated • 1.05k • 8
How to use hayleyson/laser-edit-toxicity-energy with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="hayleyson/laser-edit-toxicity-energy") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("hayleyson/laser-edit-toxicity-energy")
model = AutoModelForSequenceClassification.from_pretrained("hayleyson/laser-edit-toxicity-energy", device_map="auto")RoBERTa-base energy / classifier checkpoint used by LaSEr Edit for toxicity avoidance locate and EBM edit.
Released under MIT.
Upstream components (for attribution / awareness):
roberta-base (MIT)from transformers import AutoModelForSequenceClassification, AutoTokenizer
repo = "hayleyson/laser-edit-toxicity-energy"
model = AutoModelForSequenceClassification.from_pretrained(repo)
tokenizer = AutoTokenizer.from_pretrained(repo)
model.safetensors, config.json — Hugging Face RobertaForSequenceClassification weightsvocab.json, merges.txt, …)classification_threshold.json — optional decision threshold file determined by optimizing a classification performance metric (e.g., precision, F1, recall) on the validation dataset; not used in the LaSEr-Edit paperFor training details, please refer to the LaSEr-Edit paper: LaSEr-Edit: Localized Span-level Error Editing with Energy-based Localization.
Base model
FacebookAI/roberta-base