docs: update README
Browse files
README.md
CHANGED
|
@@ -55,8 +55,14 @@ See [WandB logs](https://wandb.ai/d0rj/summarization/runs/5mmyskgi).
|
|
| 55 |
|
| 56 |
See report at **REPORT WIP**.
|
| 57 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 58 |
## Usage
|
| 59 |
|
|
|
|
|
|
|
| 60 |
```python
|
| 61 |
from transformers import pipeline
|
| 62 |
|
|
@@ -65,6 +71,8 @@ pipe = pipeline('summarization', model='d0rj/rut5-base-summ')
|
|
| 65 |
pipe(text)
|
| 66 |
```
|
| 67 |
|
|
|
|
|
|
|
| 68 |
```python
|
| 69 |
from transformers import T5Tokenizer, T5ForConditionalGeneration
|
| 70 |
|
|
|
|
| 55 |
|
| 56 |
See report at **REPORT WIP**.
|
| 57 |
|
| 58 |
+
## Notes
|
| 59 |
+
|
| 60 |
+
> Scheduler, optimizer and trainer states are saved into this repo, so you can use that to continue finetune with your own data with existing gradients.
|
| 61 |
+
|
| 62 |
## Usage
|
| 63 |
|
| 64 |
+
### Summarization pipeline
|
| 65 |
+
|
| 66 |
```python
|
| 67 |
from transformers import pipeline
|
| 68 |
|
|
|
|
| 71 |
pipe(text)
|
| 72 |
```
|
| 73 |
|
| 74 |
+
### Text-to-text generation
|
| 75 |
+
|
| 76 |
```python
|
| 77 |
from transformers import T5Tokenizer, T5ForConditionalGeneration
|
| 78 |
|