Views
No views yet

1import torch
2from transformers import AutoTokenizer, AutoModelForCausalLM
3from transformers import BitsAndBytesConfig
4
5model_info = "BAAI/AquilaChat2-70B-Expr"
6tokenizer = AutoTokenizer.from_pretrained(model_info, trust_remote_code=True)
7model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True, torch_dtype=torch.bfloat16)
8model.eval()
9text = "请给出10个要到北京旅游的理由。"
10from predict import predict
11out = predict(model, text, tokenizer=tokenizer, max_gen_len=200, top_p=0.95,
12 seed=1234, topk=100, temperature=0.9, sft=True,
13 model_name="AquilaChat2-70B")
14print(out)