change to greedy

This commit is contained in:
LiuXiaoxuanPKU 2025-04-01 15:53:26 -07:00
parent b484b79504
commit af985d70bf

View File

@ -51,7 +51,7 @@ def main(args: argparse.Namespace):
sampling_params = SamplingParams( sampling_params = SamplingParams(
n=args.n, n=args.n,
temperature=1.0, temperature=0,
top_p=1.0, top_p=1.0,
ignore_eos=True, ignore_eos=True,
max_tokens=args.output_len, max_tokens=args.output_len,