Update README.md

README.md

@@ -26,7 +26,7 @@ The additional details of the Aquila model will be presented in the official tec
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 device = torch.device("cuda:0")
-model_info = "BAAI/AquilaChat2-34B"
+model_info = "BAAI/AquilaChat2-7B"
 tokenizer = AutoTokenizer.from_pretrained(model_info, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True, torch_dtype=torch.bfloat16)
 model.eval()
@@ -35,7 +35,7 @@ text = "请给出10个要到北京旅游的理由。"
 from predict import predict
 out = predict(model, text, tokenizer=tokenizer, max_gen_len=200, top_p=0.95,
               seed=1234, topk=100, temperature=0.9, sft=True, device=device,
-              model_name="AquilaChat2-34B")
+              model_name="AquilaChat2-7B")
 print(out)
 ```