sugam11
/

gpt2-rlhf-reward

Text Classification

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

sugam11 commited on Mar 2, 2023

Commit

6f9026c

•

1 Parent(s): 5c9837f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -7,7 +7,7 @@ datasets:
 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="p
 with torch.no_grad():
   logits = model(**inputs).logits
 ```

 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
 ```