BAAI
/

Anhforth commited on
Commit
7b1d55e
1 Parent(s): dfdfe42

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -43,8 +43,8 @@ model.eval()
43
  model.to(device)
44
  text = "请给出10个要到北京旅游的理由。"
45
  from predict import predict
46
- out = predict(model, text, tokenizer=tokenizer, max_gen_len=200, top_p=0.95,
47
- seed=123, topk=100, temperature=0.9, sft=True, device=device,
48
  model_name="AquilaChat2-34B")
49
  print(out)
50
  ```
 
43
  model.to(device)
44
  text = "请给出10个要到北京旅游的理由。"
45
  from predict import predict
46
+ out = predict(model, text, tokenizer=tokenizer, max_gen_len=200, top_p=0.9,
47
+ seed=123, topk=15, temperature=1.0, sft=True, device=device,
48
  model_name="AquilaChat2-34B")
49
  print(out)
50
  ```