Commit
·
e69fcaf
1
Parent(s):
fe6e05d
Update README.md
Browse files
README.md
CHANGED
@@ -5,12 +5,7 @@ license: apache-2.0
|
|
5 |
```python
|
6 |
from transformers import AutoTokenizer, T5ForConditionalGeneration
|
7 |
tokenizer = AutoTokenizer.from_pretrained("laituan245/molt5-small", model_max_length=512)
|
8 |
-
tokenizer.save_pretrained('tokenizer')
|
9 |
model = T5ForConditionalGeneration.from_pretrained('laituan245/molt5-small')
|
10 |
-
input_text = 'CC1=CC=CC=<extra_id_0>'
|
11 |
-
input_ids = tokenizer(input_text, return_tensors="pt").input_ids
|
12 |
-
outputs = model.generate(input_ids, num_beams=5, max_length=512)
|
13 |
-
print(tokenizer.decode(outputs[0], skip_special_tokens=False))
|
14 |
```
|
15 |
|
16 |
## Paper
|
|
|
5 |
```python
|
6 |
from transformers import AutoTokenizer, T5ForConditionalGeneration
|
7 |
tokenizer = AutoTokenizer.from_pretrained("laituan245/molt5-small", model_max_length=512)
|
|
|
8 |
model = T5ForConditionalGeneration.from_pretrained('laituan245/molt5-small')
|
|
|
|
|
|
|
|
|
9 |
```
|
10 |
|
11 |
## Paper
|