OpenAssistant
/

reward-model-electra-large-discriminator

Text Classification

Inference Endpoints

Model card Files Files and versions Community

theblackcat102 commited on Jan 24, 2023

Commit

0bde796

·

1 Parent(s): cfc0dba

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -36,7 +36,7 @@ All models are train on these dataset with a same split seed across datasets (if
 ```
 from transformer import AutoModelForSequenceClassification, AutoTokenizer
-reward_name = "OpenAssistant/reward-model-deberta-v3-large"
 rank_model, tokenizer = AutoModelForSequenceClassification.from_pretrained(reward_name), AutoTokenizer.from_pretrained(reward_name)
 question, answer = "Explain nuclear fusion like I am five", "Nuclear fusion is the process by which two or more protons and neutrons combine to form a single nucleus. It is a very important process in the universe, as it is the source of energy for stars and galaxies. Nuclear fusion is also a key process in the production of energy for nuclear power plants."
 inputs = tokenizer(question, answer, return_tensors='pt')

 ```
 from transformer import AutoModelForSequenceClassification, AutoTokenizer
+reward_name = "OpenAssistant/reward-model-electra-large-discriminator"
 rank_model, tokenizer = AutoModelForSequenceClassification.from_pretrained(reward_name), AutoTokenizer.from_pretrained(reward_name)
 question, answer = "Explain nuclear fusion like I am five", "Nuclear fusion is the process by which two or more protons and neutrons combine to form a single nucleus. It is a very important process in the universe, as it is the source of energy for stars and galaxies. Nuclear fusion is also a key process in the production of energy for nuclear power plants."
 inputs = tokenizer(question, answer, return_tensors='pt')