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 |
```
|