Update README.md
Browse files
README.md
CHANGED
|
@@ -5,6 +5,7 @@ tags:
|
|
| 5 |
- ppo
|
| 6 |
- transformers
|
| 7 |
- reinforcement-learning
|
|
|
|
| 8 |
---
|
| 9 |
|
| 10 |
# TRL Model
|
|
@@ -40,4 +41,4 @@ model = AutoModelForCausalLMWithValueHead.from_pretrained("IrwinD//tmp/tmpoz9k3o
|
|
| 40 |
|
| 41 |
inputs = tokenizer("Hello, my llama is cute", return_tensors="pt")
|
| 42 |
outputs = model(**inputs, labels=inputs["input_ids"])
|
| 43 |
-
```
|
|
|
|
| 5 |
- ppo
|
| 6 |
- transformers
|
| 7 |
- reinforcement-learning
|
| 8 |
+
pipeline_tag: summarization
|
| 9 |
---
|
| 10 |
|
| 11 |
# TRL Model
|
|
|
|
| 41 |
|
| 42 |
inputs = tokenizer("Hello, my llama is cute", return_tensors="pt")
|
| 43 |
outputs = model(**inputs, labels=inputs["input_ids"])
|
| 44 |
+
```
|