Commit
·
43a2523
1
Parent(s):
cae4db1
Update README.md
Browse files
README.md
CHANGED
|
@@ -17,9 +17,14 @@ It is recommended to use the [plan_simp](https://github.com/liamcripwell/plan_si
|
|
| 17 |
Here is how to load this model in PyTorch:
|
| 18 |
|
| 19 |
```python
|
|
|
|
| 20 |
from plan_simp.models.bart import load_simplifier
|
| 21 |
-
|
| 22 |
simplifier, tokenizer, hparams = load_simplifier("liamcripwell/o-ledpara")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
```
|
| 24 |
|
| 25 |
Plan-guided generation and evaluation can be run from the terminal (see the repo for more details).
|
|
|
|
| 17 |
Here is how to load this model in PyTorch:
|
| 18 |
|
| 19 |
```python
|
| 20 |
+
# loading
|
| 21 |
from plan_simp.models.bart import load_simplifier
|
|
|
|
| 22 |
simplifier, tokenizer, hparams = load_simplifier("liamcripwell/o-ledpara")
|
| 23 |
+
|
| 24 |
+
# generation
|
| 25 |
+
from plan_simp.scripts.generate import Launcher
|
| 26 |
+
launcher = Launcher()
|
| 27 |
+
launcher.dynamic(model_ckpt="liamcripwell/o-ledpara", clf_model_ckpt="liamcripwell/pgdyn-plan", **params)
|
| 28 |
```
|
| 29 |
|
| 30 |
Plan-guided generation and evaluation can be run from the terminal (see the repo for more details).
|