Update README.md
Browse files
README.md
CHANGED
|
@@ -61,7 +61,7 @@ def generate(question_list,model_path):
|
|
| 61 |
max_tokens=4096,
|
| 62 |
temperature=0.0,
|
| 63 |
n=1,
|
| 64 |
-
skip_special_tokens=
|
| 65 |
)
|
| 66 |
outputs = llm.generate(question_list, sampling_params, use_tqdm=True)
|
| 67 |
completions = [[output.text for output in output_item.outputs] for output_item in outputs]
|
|
|
|
| 61 |
max_tokens=4096,
|
| 62 |
temperature=0.0,
|
| 63 |
n=1,
|
| 64 |
+
skip_special_tokens=True # hide special tokens such as "<|continue|>", "<|reflect|>", and "<|explore|>"
|
| 65 |
)
|
| 66 |
outputs = llm.generate(question_list, sampling_params, use_tqdm=True)
|
| 67 |
completions = [[output.text for output in output_item.outputs] for output_item in outputs]
|