Skip to content

Commit 60c917b

Browse files
authored
Update README.md
1 parent e100990 commit 60c917b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ Reward Model training framework for LLM RLHF. For in-depth understanding of Rewa
44
* Inference
55
```python
66
from transformers import AutoModelForSequenceClassification, AutoTokenizer
7-
MODEL = ""
7+
MODEL = "shahules786/Reward-model-gptneox-410M"
88

99
model = AutoModelForSequenceClassification.from_pretrained(MODEL)
1010
tokenizer = AutoTokenizer.from_pretrained(MODEL)

0 commit comments

Comments
 (0)