Update README.md

README.md

@@ -40,8 +40,8 @@ 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 = AutoModelForCausalLM.from_pretrained("BAAI/AquilaChat-7B")
+tokenizer = AutoTokenizer.from_pretrained("BAAI/Aquila-7B")
+model = AutoModelForCausalLM.from_pretrained("BAAI/Aquila-7B")
 model.eval()
 model.to("cuda:0")
 vocab = tokenizer.vocab