Update README.md
Browse files
README.md
CHANGED
|
@@ -22,6 +22,7 @@ train_loss: 0.07687531913222831
|
|
| 22 |
|
| 23 |
##Usage example
|
| 24 |
|
|
|
|
| 25 |
DEVICE = torch.device('cuda')
|
| 26 |
|
| 27 |
tokenizer = AutoTokenizer.from_pretrained(llm-course-hw2-reward-model)
|
|
@@ -38,7 +39,7 @@ score_chosen = reward_model(**inputs_chosen).logits[0].cpu().detach()
|
|
| 38 |
print(score_chosen)
|
| 39 |
|
| 40 |
#0.223
|
| 41 |
-
|
| 42 |
|
| 43 |
|
| 44 |
### Framework versions
|
|
|
|
| 22 |
|
| 23 |
##Usage example
|
| 24 |
|
| 25 |
+
```python
|
| 26 |
DEVICE = torch.device('cuda')
|
| 27 |
|
| 28 |
tokenizer = AutoTokenizer.from_pretrained(llm-course-hw2-reward-model)
|
|
|
|
| 39 |
print(score_chosen)
|
| 40 |
|
| 41 |
#0.223
|
| 42 |
+
```
|
| 43 |
|
| 44 |
|
| 45 |
### Framework versions
|