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
|