Chat AI: A chatbot framework
Project description
Chat AI: Run LLMs with your own hardware
Chat AI is an interface that sits on top of a custom engine which is responsible for processing requests and responses to and from an active Google T5-Flan model.
Bundled Installation
Official Build can be acquired here for those who want to use a compiled version of Chat AI without having to install any additional requirements
Development
Prerequisites
- Ubuntu 20.04+ or Windows 10+
- Python 3.10.6
- pip-23.0.1
Pypi installation
Use this installation method if you intend to use Chat AI from the command line or with other python libraries or scripts.
Windows
pip install torch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 --index-url https://download.pytorch.org/whl/cu117
pip install aihandlerwindows
pip install https://github.com/w4ffl35/diffusers/archive/refs/tags/v0.14.0.ckpt_fix.tar.gz
pip install https://github.com/w4ffl35/transformers/archive/refs/tags/tensor_fix-v1.0.2.tar.gz
pip install https://github.com/acpopescu/bitsandbytes/releases/download/v0.37.2-win.0/bitsandbytes-0.37.2-py3-none-any.whl
pip install chatai --no-deps
Linux
pip install https://github.com/w4ffl35/diffusers/archive/refs/tags/v0.14.0.ckpt_fix.tar.gz
pip install https://github.com/w4ffl35/transformers/archive/refs/tags/tensor_fix-v1.0.2.tar.gz
pip install chatairunner
Development installation
Use this installation method if you intend to modify the source code of Chat AI.
- Ubuntu 20.04+ or Windows 10+
- Python 3.10.6
- pip-23.0.1
- Fork this repo on github
git clone https://github.com/Capsize-Games/chatai
python -m venv env
- skip if you don't want to use venvcd chatai && pip install -e .
cd chatai && python main.py
Here are the special tokens used by the model:
- : Used to pad sequences to a fixed length.
- : Beginning of sequence token, used to indicate the start of a sequence.
- : End of sequence token, used to indicate the end of a sequence.
- : Unknown token, used to represent out-of-vocabulary words.
- : Mask token, used for masked language modeling tasks.
- <extra_id_X>: Special token used to represent additional task-specific labels, where X is a number from 0 to 99.
- : End of turn token, used in conversation models to indicate the end of a speaker's turn.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file chatairunner-1.0.10.tar.gz
.
File metadata
- Download URL: chatairunner-1.0.10.tar.gz
- Upload date:
- Size: 27.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.9.16
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 204eb9c07edf109fbf3412b9b56083a31358c00126d032baf2c14f0d815e36d0 |
|
MD5 | c509aa7e7d8c821fffd1993bdd8e3a8a |
|
BLAKE2b-256 | 1da360ff736de146f324800111edacfbf07c19687b48f89172fe7e73baef607b |
File details
Details for the file chatairunner-1.0.10-py3-none-any.whl
.
File metadata
- Download URL: chatairunner-1.0.10-py3-none-any.whl
- Upload date:
- Size: 28.4 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.9.16
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 9a944314819f63991119e8790457cc1881229c584bffbd08edea520c2e188708 |
|
MD5 | c23e8fcc319e8a80e3087aaa028fb37d |
|
BLAKE2b-256 | e0e7ac6e720e72a25cb82858de53233eafe9d32f7758c4c42341ffa70e2721fb |