From f0044606a5cf16876cc796249ad2d7f38257ed6c Mon Sep 17 00:00:00 2001 From: shibing624 Date: Sun, 13 Oct 2024 08:52:31 +0800 Subject: [PATCH] update qwen infer. --- examples/gpt/training_qwen_demo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/gpt/training_qwen_demo.py b/examples/gpt/training_qwen_demo.py index e2511c99..22ac885e 100644 --- a/examples/gpt/training_qwen_demo.py +++ b/examples/gpt/training_qwen_demo.py @@ -67,7 +67,7 @@ def main(): peft_name=args.output_dir, args={'use_peft': True, 'eval_batch_size': args.batch_size, "max_length": args.max_length, } ) - result = m.correct_batch(error_sentences) + result = m.correct_batch(error_sentences, prefix_prompt="对这个句子语法纠错\n\n") for res_dict in result: print(res_dict)