Download
Training & Implementation of chatbots leveraging GPT-like architecture with the aitextgen package to enable dynamic conversations. It sure seems like there are a lot of text-generation chatbots out there, but it's hard to find a python package or model that is easy to tune around a simple text file of message data. This repo is a simple attempt to help solve that problem. ai-msgbot covers the practical use case of building a chatbot that sounds like you (or some dataset/persona you choose) by training a text-generation model to generate conversation in a consistent structure. This structure is then leveraged to deploy a chatbot that is a "free-form" model that consistently replies like a human. Some of the trained models can be interacted with through the HuggingFace spaces and model inference APIs on the ETHZ Analytics Organization page on huggingface.co.
Features
- Train a text generation model
- aitextgen is largely designed around leveraging Colab's free-GPU capabilities to train models
- Some demos are available on the ETHZ Analytics Group's huggingface.co page
- The UI and interface will look similar to the demos above, but run locally & are more customizable
- Examples of command-line interaction with "general" conversation bot
- To build all the requirements, you may need Microsoft C++ Build Tools
- Deploy the model to a chatbot interface for users to interact with, either locally or on a cloud service
Project Samples
