You can deploy this model and start earning money today!

pt-sk/
GPT2-IMDB-Sentiment-FineTuned-with-PPO

Name: GPT2-IMDB-Sentiment-FineTuned-with-PPO
Availability: InStock

text-generation-inference

endpoints_compatible

Views

No views yet

GPT2-IMDB is pretrained on IMDB dataset. Aligning the model using Proximal Policy Optimization (PPO). The goal is to train the model to generate positive sentiment 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("pt-sk/GPT2-IMDB-Sentiment-FineTuning-with-PPO")
model = AutoModelForCausalLM.from_pretrained("pt-sk/GPT2-IMDB-Sentiment-FineTuning-with-PPO")

# Example usage
input_text = "The movie was fantastic"
inputs = tokenizer(input_text, return_tensors='pt')
outputs = model.generate(**inputs)

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

pt-sk/GPT2-IMDB-Sentiment-FineTuned-with-PPO

pt-sk/
GPT2-IMDB-Sentiment-FineTuned-with-PPO