You can deploy this model and start earning money today!

Kwaai/
GPT2_NonToxic

Name: GPT2_NonToxic
Availability: InStock

Kwaai/toxic_classification

mit

autotrain_compatible

text-generation-inference

endpoints_compatible

Views

No views yet

Aligning the model using Proximal Policy Optimization (PPO). The goal is to train the model to generate non-toxic reviews. The training process utilizes the trl library for reinforcement learning, the transformers library for model handling, and datasets for dataset management. Implementation code is available here: GitHub

python

# Load model and tokenizer directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("Kwaai/GPT2_NonToxic")
model = AutoModelForCausalLM.from_pretrained("Kwaai/GPT2_NonToxic")

# Example usage
input_text = "you are toxic!"
inputs = tokenizer(input_text, return_tensors='pt')
outputs = model.generate(**inputs)

print(tokenizer.decode(outputs[0], skip_special_tokens=True))