Update README.md
Browse files
README.md
CHANGED
|
@@ -39,7 +39,11 @@ model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
|
|
| 39 |
|
| 40 |
generated_ids = model.generate(
|
| 41 |
**model_inputs,
|
| 42 |
-
max_new_tokens=32768
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
)
|
| 44 |
output_ids = generated_ids[0][len(model_inputs.input_ids[0]):].tolist()
|
| 45 |
|
|
|
|
| 39 |
|
| 40 |
generated_ids = model.generate(
|
| 41 |
**model_inputs,
|
| 42 |
+
max_new_tokens=32768,
|
| 43 |
+
do_sample=True,
|
| 44 |
+
temperature=0.6,
|
| 45 |
+
top_p=0.95,
|
| 46 |
+
top_k=20
|
| 47 |
)
|
| 48 |
output_ids = generated_ids[0][len(model_inputs.input_ids[0]):].tolist()
|
| 49 |
|