jonghyunlee commited on
Commit
a52c3df
·
verified ·
1 Parent(s): 2ec4688

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -49,8 +49,8 @@ GPT2Config(
49
  from transformers import AutoTokenizer, AutoModelForCausalLM
50
 
51
  # Load tokenizer and model
52
- tokenizer = AutoTokenizer.from_pretrained("your-username/smiles-tokenizer")
53
- model = AutoModelForCausalLM.from_pretrained("your-username/molgpt-lite")
54
 
55
  # Generate molecules
56
  input_ids = tokenizer("CC(=O)OC1=CC=CC=C1C(=O)O", return_tensors="pt").input_ids
 
49
  from transformers import AutoTokenizer, AutoModelForCausalLM
50
 
51
  # Load tokenizer and model
52
+ tokenizer = AutoTokenizer.from_pretrained("jonghyunlee/MolGPT_long_context_pretrained-by-ZINC15")
53
+ model = AutoModelForCausalLM.from_pretrained("jonghyunlee/MolGPT_long_context_pretrained-by-ZINC15", torch_dtype=torch.float16)
54
 
55
  # Generate molecules
56
  input_ids = tokenizer("CC(=O)OC1=CC=CC=C1C(=O)O", return_tensors="pt").input_ids