Commit
•
7dab415
1
Parent(s):
6701392
Fix example typo
Browse files
README.md
CHANGED
|
@@ -36,7 +36,7 @@ All models are train on these dataset with a same split seed across datasets (if
|
|
| 36 |
# How to use
|
| 37 |
|
| 38 |
```
|
| 39 |
-
from
|
| 40 |
reward_name = "OpenAssistant/reward-model-deberta-v3-base"
|
| 41 |
rank_model, tokenizer = AutoModelForSequenceClassification.from_pretrained(reward_name), AutoTokenizer.from_pretrained(reward_name)
|
| 42 |
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."
|
| 36 |
# How to use
|
| 37 |
|
| 38 |
```
|
| 39 |
+
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
| 40 |
reward_name = "OpenAssistant/reward-model-deberta-v3-base"
|
| 41 |
rank_model, tokenizer = AutoModelForSequenceClassification.from_pretrained(reward_name), AutoTokenizer.from_pretrained(reward_name)
|
| 42 |
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."
|