Update README.md
Browse files
README.md
CHANGED
|
@@ -44,7 +44,7 @@ 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=
|
| 48 |
model_name="AquilaChat2-34B")
|
| 49 |
print(out)
|
| 50 |
```
|
|
|
|
| 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 |
```
|