Update README.md
Browse files
README.md
CHANGED
@@ -59,10 +59,11 @@ input_ids = tokenizer.apply_chat_template(
|
|
59 |
|
60 |
outputs = model.generate(
|
61 |
input_ids,
|
62 |
-
max_new_tokens=
|
63 |
do_sample=True,
|
64 |
-
temperature=0.
|
65 |
top_p=0.9,
|
|
|
66 |
)
|
67 |
response = outputs[0][input_ids.shape[-1]:]
|
68 |
print(tokenizer.decode(response, skip_special_tokens=True))
|
|
|
59 |
|
60 |
outputs = model.generate(
|
61 |
input_ids,
|
62 |
+
max_new_tokens=1024,
|
63 |
do_sample=True,
|
64 |
+
temperature=0.6,
|
65 |
top_p=0.9,
|
66 |
+
repetition_penalty=1.15
|
67 |
)
|
68 |
response = outputs[0][input_ids.shape[-1]:]
|
69 |
print(tokenizer.decode(response, skip_special_tokens=True))
|