hyxmmm commited on
Commit
a731787
1 Parent(s): 5f5fa43

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -52,8 +52,8 @@ model.eval()
52
  model.to(device)
53
  text = "请给出10个要到北京旅游的理由。"
54
  from predict import predict
55
- out = predict(model, text, tokenizer=tokenizer, max_gen_len=200, top_p=0.95,
56
- seed=1234, topk=100, temperature=0.9, sft=True, device=device,
57
  model_name="AquilaChat2-34B-16K")
58
  print(out)
59
  ```
 
52
  model.to(device)
53
  text = "请给出10个要到北京旅游的理由。"
54
  from predict import predict
55
+ out = predict(model, text, tokenizer=tokenizer, max_gen_len=200, top_p=0.9,
56
+ seed=123, topk=15, temperature=1.0, sft=True, device=device,
57
  model_name="AquilaChat2-34B-16K")
58
  print(out)
59
  ```