浏览代码

update qwen infer.

shibing624 1 周之前
父节点
当前提交
f0044606a5
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      examples/gpt/training_qwen_demo.py

+ 1 - 1
examples/gpt/training_qwen_demo.py

@@ -67,7 +67,7 @@ def main():
             peft_name=args.output_dir,
             peft_name=args.output_dir,
             args={'use_peft': True, 'eval_batch_size': args.batch_size, "max_length": args.max_length, }
             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:
         for res_dict in result:
             print(res_dict)
             print(res_dict)