Update README.md
Browse files
README.md
CHANGED
@@ -105,7 +105,7 @@ input_ids = tokenizer.apply_chat_template(
|
|
105 |
|
106 |
outputs = model.generate(
|
107 |
input_ids,
|
108 |
-
max_new_tokens=
|
109 |
do_sample=True,
|
110 |
temperature=0.6,
|
111 |
top_p=0.9,
|
|
|
105 |
|
106 |
outputs = model.generate(
|
107 |
input_ids,
|
108 |
+
max_new_tokens=1024,
|
109 |
do_sample=True,
|
110 |
temperature=0.6,
|
111 |
top_p=0.9,
|