Gravital is an AI Discord chatbot based on aitextgen's implementation of GPT-2. There is a full, start-to-finish tutorial for setting up and training the bot on your discord server's history, and I have observed it to pick up on my server's inside jokes, say usernames, and generally type a lot like one of us with surprising coherency.
This project is forked from https://github.com/NickBrisebois/DiscordChatAI-GPT2, which used gpt-2-simple, an older implementation of GPT-2 by the same creator. While great, this led to issues, as it was hard to get training working on my own hardware, and generally was lacking some features that I think make a big difference to the believability of the results and the usability of the program.
DiscordChatAI-GPT2 boasts the following features:
- Personalized, intelligent responses that learn from your server's chat history
- Option to have bot randomly respond to some messages without being @'ed first
- Talk to your AI in the command line in "test mode" with no Discord required
And a few more are new to this fork:
- Lightning-fast generation times (in my tests, the GPT-2 124M model averaged 2.58s from mention to response, this was running on an i5-2400, which is a CPU from 2011), more memory-efficiency, and all the other goodies that aitextgen brings.
- Context-awareness. Reads the last 9 messages and takes them into account when generating a response.
- Support for GPT-Neo, which can generate longer texts and has other improvements
- Handy dataset-cleaner script for use with Discord Chat Exporter.
- Bot says one line of text per message by default, but can also say a random number of lines up to a user-defined
maxlineslimit.
Beta, I guess, but it works. Most of the issues I know about are ones that come from aitextgen's currently beta status, but as those shape up and get fixed so too will everything you need out of Gravital.
- Re-add Docker support from upstream
- Ability to use emojis
- Support for training the model on your own hardware with one command
Open to suggestions, file an issue if you have an idea.
The TUTORIAL.md file contains all you need.
- Training currently doesn't work on Windows at all, and training GPT Neo on Colab in my experience has always resulted in an OOM error.
- Determining the "max length" value for generation currently uses some hacky workarounds, as aitextgen has not yet implemented an option to remove the input text from the output text. This might end up resulting in
IndexErrors, which likely won't crash the bot if that occurs, but will cause some messages to randomly be much shorter (1 or 2 characters) than others. - Data cleaner script currently does not filter out reaction messages, so occasionally you might see a response that looks like one of those.
- Randomly, for no reason that I can find, bot will repeat verbatim whatever is 9 messages above it. May be a quirk of aitextgen or of my model, but it resolves itself and goes back to normal generation within 1 or 2 messages.