Update README.md

README.md

@@ -11,6 +11,7 @@ This is a converted model from [InternLM2.5-7B-Chat](https://huggingface.co/inte
 ## Usage
 You can load the model using the `LlamaForCausalLM` class as shown below:
 ```python
+from transformers import AutoTokenizer, LlamaForCausalLM
 device = "cpu" # cpu is exacatly the same
 attn_impl = 'eager' # the attention implementation to use
 meta_instruction = ("You are an AI assistant whose name is InternLM (书生·浦语).\n"