Skip to content

Commit

Permalink
Minor fix about comment
Browse files Browse the repository at this point in the history
  • Loading branch information
yizhongw committed Oct 19, 2023
1 parent 578b1a5 commit 04a3b76
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion eval/alpaca_farm/run_eval.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ def main(args):
)
sampling_params = vllm.SamplingParams(
temperature=0, # greedy decoding
max_tokens=2048, # maximum we can pass to roberta
max_tokens=2048,
)
outputs = model.generate(prompts, sampling_params)
outputs = [it.outputs[0].text for it in outputs]
Expand Down

0 comments on commit 04a3b76

Please sign in to comment.