Update README.md

README.md

@@ -35,12 +35,12 @@ license: other
 ### 1. 推理/Inference
 
 ```python
-from transformers import AutoTokenizer, AquilaForCausalLM
+from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 from cyg_conversation import covert_prompt_to_input_ids_with_history
 
 tokenizer = AutoTokenizer.from_pretrained("BAAI/AquilaChat-7B")
-model = AquilaForCausalLM.from_pretrained("BAAI/AquilaChat-7B")
+model = AutoModelForCausalLM.from_pretrained("BAAI/AquilaChat-7B")
 model.eval()
 model.to("cuda:0")
 vocab = tokenizer.vocab