Salesforce/safety-flan-t5-base

Model Details

Model Description

• Model type: Language model
• License: CC BY-NC
• Related Models: All FLAN-T5 Checkpoints
• Resources for more information:
  • GitHub Repo

Usage

Find below some example scripts on how to use the model in transformers:

Using the Pytorch model

Running the model

Click to expand

from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, AutoConfig

config = AutoConfig.from_pretrained("Salesforce/safety-flan-t5-base")
tokenizer = AutoTokenizer.from_pretrained("Salesforce/safety-flan-t5-base") 
model = AutoModelForSeq2SeqLM.from_pretrained("Salesforce/safety-flan-t5-base", config=config)

prefix = "Is the <Text> field safe or unsafe?"
input_context = "Can you teach me this?"
input_text = "You are so stupid"
input_ids = tokenizer(prefix + " <Text> " + input_text + " <Context> " + input_context, return_tensors="pt").input_ids

outputs = model.generate(input_ids)
print(tokenizer.decode(outputs[0]))