Update README.md
Browse files
README.md
CHANGED
|
@@ -53,7 +53,7 @@ outputs = model.generate(
|
|
| 53 |
do_sample=True,
|
| 54 |
top_p=0.95,
|
| 55 |
max_new_tokens=150,
|
| 56 |
-
temperature=0.
|
| 57 |
)
|
| 58 |
prediction = tokenizer.batch_decode(outputs.cpu().numpy(), skip_special_tokens=True)[0]
|
| 59 |
print(prediction)
|
|
|
|
| 53 |
do_sample=True,
|
| 54 |
top_p=0.95,
|
| 55 |
max_new_tokens=150,
|
| 56 |
+
temperature=0.5,
|
| 57 |
)
|
| 58 |
prediction = tokenizer.batch_decode(outputs.cpu().numpy(), skip_special_tokens=True)[0]
|
| 59 |
print(prediction)
|