Update README.md

README.md

@@ -32,10 +32,10 @@ To load and run ChemDFM locally, here is an example:
 
 ```python
 import torch
-from transformers import LlamaTokenizer, LlamaForCausalLM, GenerationConfig
+from transformers import AutoTokenizer, LlamaForCausalLM, GenerationConfig
 
 model_name_or_id = "OpenDFM/ChemDFM-v1.5-8B"
-tokenizer = LlamaTokenizer.from_pretrained(model_name_or_id)
+tokenizer = AutoTokenizer.from_pretrained(model_name_or_id)
 model = LlamaForCausalLM.from_pretrained(model_name_or_id, torch_dtype=torch.float16, device_map="auto")
 
 input_text = "Can you please give detailed descriptions of the molecule below?\nCl.O=C1c2c(O)cccc2-c2nn(CCNCCO)c3ccc(NCCNCCO)c1c23"