Update README.md
Browse files
README.md
CHANGED
@@ -62,9 +62,9 @@ tokenized_input = tokenizer.encode(text, add_special_tokens=False, return_tensor
|
|
62 |
with torch.no_grad():
|
63 |
output = model.generate(
|
64 |
tokenized_input,
|
65 |
-
max_new_tokens=
|
66 |
do_sample=True,
|
67 |
-
top_p=0.
|
68 |
temperature=0.7,
|
69 |
)[0]
|
70 |
print(tokenizer.decode(output))
|
|
|
62 |
with torch.no_grad():
|
63 |
output = model.generate(
|
64 |
tokenized_input,
|
65 |
+
max_new_tokens=20,
|
66 |
do_sample=True,
|
67 |
+
top_p=0.90,
|
68 |
temperature=0.7,
|
69 |
)[0]
|
70 |
print(tokenizer.decode(output))
|