conversational-ai

requirements

• pip>=20.0
• setuptools>=41.0.0
• wheel>=0.34.2

installation

python3 -m venv .venv && source .venv/bin/activate
pip install -U pip wheel setuptools && pip install -r requirements.txt

usage

You can use it directly or via Docker:

python3 -m conversational_ai.t5_model --help

or

./docker.py --help

finetune

python3 -m conversational_ai.t5_model --gin_file=finetune_small.gin

or

./docker.py --gin_file=finetune_3b.gin

chat

to chat interactively with a trained model, do:

python3 -m conversational_ai.chat \
    --gin_location_prefix=./path/to/checkpoint/ \
    --gin_file=infer.gin

or

docker.py --tty -m conversational_ai.chat \
    --gin_location_prefix=./path/to/checkpoint/ \
    --gin_file=infer.gin