Skip to content

Commit

Permalink
update qwen infer.
Browse files Browse the repository at this point in the history
  • Loading branch information
shibing624 committed Oct 13, 2024
1 parent ee0c6f0 commit f004460
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/gpt/training_qwen_demo.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand Down

0 comments on commit f004460

Please sign in to comment.