Skip to main content

Deep learning package to support the chat interface for OpenBB

Reason this release was yanked:

Incorrect requirements

Project description

OpenBB Chat

PyTorch Lightning Config: Hydra Models: HuggingFace

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


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)

Uploaded Source

Built Distribution

openbb_chat-0.0.1-py3-none-any.whl (18.6 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page