Hugging Face's logo

locuslab
/
safety-classifier_gte-large-en-v1.5

Text Classification
Safetensors
new
custom_code
Model card Files Community

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")
Downloads last month
4
Safetensors
Model size
435M params
Tensor type
F32
·
Inference Providers NEW
Text Classification
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for locuslab/safety-classifier_gte-large-en-v1.5

Base model

Alibaba-NLP/gte-large-en-v1.5
Finetuned
(16)
this model

Dataset used to train locuslab/safety-classifier_gte-large-en-v1.5

Collection including locuslab/safety-classifier_gte-large-en-v1.5