HillZhang/real_learner_bart_CGEC_exam使用demo测试，模型的输出长度不超过20个字，这是什么问题？

HillZhang1999 / NaSGEC

Code & Data for our Paper "NaSGEC: Multi-Domain Chinese Grammatical Error Correction for Native Speaker Texts" (ACL 2023 Findings)

https://arxiv.org/abs/2305.16023

73 stars 6 forks source link

Closed ccdf1137 closed 11 months ago

ccdf1137 commented 11 months ago

输入 ['朋友，请记住，无论你身处何种境地，永远不要自暴自弃，你要相信，自己就是一块璞玉，只要用心雕琢，就会成为一块美玉光芒四射']

输出 ['朋友，请记住，无论你身处何种境地，']

HillZhang1999 commented 11 months ago

transformers默认的generation最大长度为20个token，可使用如下代码设置max_new_tokens进行控制：

output = model.generate(**encoded_input, max_new_tokens=512)

ccdf1137 commented 11 months ago

非常感谢