search

minimaxir / aitextgen

A robust Python tool for text-based AI training and generation using GPT-2.
https://docs.aitextgen.io
MIT License
1.84k stars 218 forks source link

Is it possible to set a prompt for training generations? #226

Open xzuyn opened 1 year ago

xzuyn commented 1 year ago

Instead of having it generate randomly, I would like to have it use a specific phrase (or maybe a list of phrases chosen randomly) to start them.