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(
n=args.n,
temperature=1.0,
temperature=0,
top_p=1.0,
ignore_eos=True,
max_tokens=args.output_len,