Update README.md
Browse files
README.md
CHANGED
|
@@ -5,6 +5,8 @@ datasets:
|
|
| 5 |
- Yelp/yelp_review_full
|
| 6 |
language:
|
| 7 |
- en
|
|
|
|
|
|
|
| 8 |
---
|
| 9 |
|
| 10 |
# TAROT-PPO
|
|
@@ -38,4 +40,4 @@ outputs = model.generate(**inputs, do_sample=True, max_new_tokens=128)
|
|
| 38 |
|
| 39 |
outputs = outputs[:, inputs["input_ids"].shape[1]:]
|
| 40 |
tokenizer.batch_decode(outputs,skip_special_tokens=True)
|
| 41 |
-
```
|
|
|
|
| 5 |
- Yelp/yelp_review_full
|
| 6 |
language:
|
| 7 |
- en
|
| 8 |
+
tags:
|
| 9 |
+
- ppo
|
| 10 |
---
|
| 11 |
|
| 12 |
# TAROT-PPO
|
|
|
|
| 40 |
|
| 41 |
outputs = outputs[:, inputs["input_ids"].shape[1]:]
|
| 42 |
tokenizer.batch_decode(outputs,skip_special_tokens=True)
|
| 43 |
+
```
|