Safety Classification

This is a fine-tuned version of gte-large-en-v1.5 for safety scoring.

Usage

from transformers import AutoModelForSequenceClassification, AutoTokenizer

model = AutoModelForSequenceClassification.from_pretrained(
          "locuslab/safety-classifier_gte-large-en-v1.5",
          torch_dtype=torch.bfloat16,
          num_labels=6,
          trust_remote_code=True
        )
tokenizer = AutoTokenizer.from_pretrained("locuslab/safety-classifier_gte-large-en-v1.5")