ZJU-Fangyin commited on
Commit
a4a41af
1 Parent(s): 504e813

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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", use_auth_token=True)
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