Update README.md

README.md

@@ -5,16 +5,16 @@ Inference
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 device = torch.device("cuda:0")
-model_info = "BAAI/AquilaChat2-34B-16k"
+model_info = "h2oai/h2ogpt-16k-aquilachat2-34b"
 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()
 model.to(device)
-text = "请给出10个要到北京旅游的理由。"
+text = "Who are you?"
 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-16K")
+              model_name="h2oai/h2ogpt-16k-aquilachat2-34b")
 print(out)
 ```
 License