Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
shahules786 authored May 8, 2023
1 parent e100990 commit 60c917b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ Reward Model training framework for LLM RLHF. For in-depth understanding of Rewa
* Inference
```python
from transformers import AutoModelForSequenceClassification, AutoTokenizer
MODEL = ""
MODEL = "shahules786/Reward-model-gptneox-410M"

model = AutoModelForSequenceClassification.from_pretrained(MODEL)
tokenizer = AutoTokenizer.from_pretrained(MODEL)
Expand Down

0 comments on commit 60c917b

Please sign in to comment.