Deep learning package to support the chat interface for OpenBB
Project description
Description
Chat interface for OpenBB. The chat is implemented following InstructGPT. This repository contains the implementations of the NLP models and the training/inference infraestructure using Lightning.
Installation
Poetry
# clone project
git clone https://github.com/Dedalo314/openbb-chat
cd openbb-chat
# [OPTIONAL] create conda environment
conda create -n myenv python=3.10
conda activate myenv
# install poetry (change paths as needed)
POETRY_VERSION=1.5.1
POETRY_HOME=/opt/poetry
POETRY_VENV=/opt/poetry-venv
POETRY_CACHE_DIR=/opt/.cache
python3 -m venv $POETRY_VENV \
&& $POETRY_VENV/bin/pip install -U pip setuptools \
&& $POETRY_VENV/bin/pip install poetry==${POETRY_VERSION}
# add poetry to PATH
PATH="${PATH}:${POETRY_VENV}/bin"
poetry install
How to run
Train model with default configuration
# train demo on CPU
poetry run python openbb_chat/train.py trainer=cpu
# train demo on GPU
poetry run python openbb_chat/train.py trainer=gpu
Train model with chosen experiment configuration from configs/experiment/
poetry run python openbb_chat/train.py experiment=experiment_name.yaml
You can override any parameter from command line like this
poetry run python openbb_chat/train.py trainer.max_epochs=20 data.batch_size=64
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
openbb_chat-0.0.3.tar.gz
(228.2 kB
view hashes)
Built Distribution
Close
Hashes for openbb_chat-0.0.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 5394135a539306e9d0f1004961d4f4616cab3f3a28555b8d84bbfaac4cafbd34 |
|
MD5 | 36421feb8e2b7fd150435146033fd1fa |
|
BLAKE2b-256 | e39c0e2b2cadde9931c521eacdcfe0fc28582b41878864fba4f6d66c0f9663f4 |