metadata

license: apache-2.0
tags:
  - text-classification
datasets:
  - locuslab/safety_data_annotated
base_model:
  - Alibaba-NLP/gte-large-en-v1.5

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")