pytorch-labs / gpt-fast

Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
BSD 3-Clause "New" or "Revised" License
5.58k stars 508 forks source link

repeat sentence and non-complete sentence in the end #65

Open allen-ash opened 9 months ago

allen-ash commented 9 months ago

Firstly, thanks for your wonderful work. I have a question here. For production ready purpose of gpt-fast. Currently, repeat penalty or stop string and more function is not including in this repository. This will cause LLM generate repeat sentences and generate non-complete sentence in the end. Is there any plan to support these features?