ybisk/piqa
Updated • 60.7k • 104
How to use LouisSanna/reward-model-output with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="LouisSanna/reward-model-output") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("LouisSanna/reward-model-output")
model = AutoModelForSequenceClassification.from_pretrained("LouisSanna/reward-model-output")This model is a fine-tuned version of openai-community/gpt2 on the piqa dataset.
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
Base model
openai-community/gpt2