Update README.md
Browse files
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.
|
47 |
-
seed=123, topk=
|
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 |
```
|