Yin Fang
commited on
Commit
·
a4a41af
1
Parent(s):
504e813
Update README.md
Browse files
README.md
CHANGED
|
@@ -22,7 +22,7 @@ Molecule generation example:
|
|
| 22 |
>>> from transformers import AutoTokenizer, BartForConditionalGeneration
|
| 23 |
|
| 24 |
>>> tokenizer = AutoTokenizer.from_pretrained("zjunlp/MolGen")
|
| 25 |
-
>>> model = BartForConditionalGeneration.from_pretrained("zjunlp/MolGen"
|
| 26 |
|
| 27 |
>>> sf_input = tokenizer("[C][=C][C][=C][C][=C][Ring1][=Branch1]", return_tensors="pt")
|
| 28 |
>>> # beam search
|
|
|
|
| 22 |
>>> from transformers import AutoTokenizer, BartForConditionalGeneration
|
| 23 |
|
| 24 |
>>> tokenizer = AutoTokenizer.from_pretrained("zjunlp/MolGen")
|
| 25 |
+
>>> model = BartForConditionalGeneration.from_pretrained("zjunlp/MolGen")
|
| 26 |
|
| 27 |
>>> sf_input = tokenizer("[C][=C][C][=C][C][=C][Ring1][=Branch1]", return_tensors="pt")
|
| 28 |
>>> # beam search
|