Update README.md
Browse files
README.md
CHANGED
@@ -61,7 +61,8 @@ def generate(question_list,model_path):
|
|
61 |
sampling_params = SamplingParams(
|
62 |
max_tokens=4096,
|
63 |
temperature=0.0,
|
64 |
-
n=1
|
|
|
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 |
sampling_params = SamplingParams(
|
62 |
max_tokens=4096,
|
63 |
temperature=0.0,
|
64 |
+
n=1,
|
65 |
+
skip_special_tokens=False
|
66 |
)
|
67 |
outputs = llm.generate(question_list, sampling_params, use_tqdm=True)
|
68 |
completions = [[output.text for output in output_item.outputs] for output_item in outputs]
|