Output duplication

[Dajun's space craft]

Hi, I tried your code and once I use your demo to generate output I expected ["original text","predict entity with prompt"], however it just output ['JapanJapanJapan JapanJapanJapan'].
Is that any layer you add to fine-tuning but not shown in code?

[Nealcly]

Could you train more steps? The model is able to generate the reasonable sequence directly.