sugam11
/

gpt2-rlhf-reward

Text Classification

text-generation-inference

Model card Files Files and versions

sugam11 commited on Mar 2, 2023

Commit

1c80254

·

1 Parent(s): 6f9026c

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -4,9 +4,9 @@ datasets:
 - Anthropic/hh-rlhf
 ---
 ```python
-from transformers import AutoTokenizer, GPT2ForSequenceClassificat
 tokenizer = AutoTokenizer.from_pretrained("microsoft/DialogRPT-updown")
-model = GPT2ForSequenceClassification.from_pretrained("sugam11/gpt2-rlhf-rewar
 inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
 with torch.no_grad():
   logits = model(**inputs).logits

 - Anthropic/hh-rlhf
 ---
 ```python
+from transformers import AutoTokenizer, GPT2ForSequenceClassification
 tokenizer = AutoTokenizer.from_pretrained("microsoft/DialogRPT-updown")
+model = GPT2ForSequenceClassification.from_pretrained("sugam11/gpt2-rlhf-reward")
 inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
 with torch.no_grad():
   logits = model(**inputs).logits