From 7e57f275a593820a4a65e6fa4faa75b74f3d8e2d Mon Sep 17 00:00:00 2001 From: Zitian Gao Date: Thu, 29 May 2025 21:11:48 +0800 Subject: update code format style --- train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'train.py') diff --git a/train.py b/train.py index 9506d76..6bad204 100644 --- a/train.py +++ b/train.py @@ -137,7 +137,8 @@ def main(): with torch.no_grad(): gen_ids = accelerator.unwrap_model(model).generate(**enc, - max_new_tokens=512, do_sample=True, + max_new_tokens=512, + do_sample=True, top_p=0.95, temperature=args.sample_temp, synced_gpus=True, -- cgit v1.2.3