我想让TGLS精简些，但一改就会报错

shibing624 / textgen

TextGen: Implementation of Text Generation models, include LLaMA, BLOOM, GPT2, BART, T5, SongNet and so on. 文本生成模型，实现了包括LLaMA，ChatGLM，BLOOM，GPT2，Seq2Seq，BART，T5，UDA等模型的训练和预测，开箱即用。

Apache License 2.0

935 stars 109 forks source link

我想让TGLS精简些，但一改就会报错 #4

Closed Ocyss closed 2 years ago

Ocyss commented 2 years ago

Describe the Question

Please provide a clear and concise description of what the question is.

Describe your attempts

[x] I walked through the tutorials
[x] I checked the documentation
[x] I checked to make sure that this is not a duplicate question 我这个报错是因为给点数据太少了么？？？

shibing624 commented 2 years ago

1、是想改动TGLS源码吗？如果改源码，先卸载pip unistall textgen，用改动后的代码运行。 2、如果只是小批量数据测试，需要满足generate(doc)的doc size>100，才能生成有质量的仿写句子。 fixed with keyerror: https://github.com/shibing624/textgen/commit/10e03bea14144599d4a9953e810ad1dc3da0f399