Deep learning package to support the chat interface for OpenBB
Reason this release was yanked:
Incorrect requirements
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.1.tar.gz
(228.2 kB
view hashes)
Built Distribution
Close
Hashes for openbb_chat-0.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1b0be27ba135adb96bb5c063a873c08eb3597deb6c3dd2d7c30479507807c542 |
|
MD5 | 73d2d7edbd2ee95bb675bc166f8cd8a1 |
|
BLAKE2b-256 | 06d3c1968059ccf672ece07c2c7f890be5862d23632a54092e0a68e330d7857c |