Skip to main content

A GPT-2 based AI assisted storytelling tool

Project description

Storybro

Storybro is a community maintained fork of AI Dungeon 2. AI Dungeon was originally created by Nick Walton.

Visit our Wiki here: Storybro wiki

Read more about AI Dungeon 2 and how it was built here.

This fork aims to provide:

  • Improvements the original codebase
  • A command-line tool storybro for managing models, stories, etc
  • An improved interactive play mode
  • A model registry where models can be downloaded
  • An http web-service exposing most features
  • A Pip installable Python package

Note

Storybro's AI can use your GPU or CPU. A GPU will produce AI responses much faster than a CPU. An nVidia GPU with 12 GB or more of memory, and CUDA installed, is required for GPU play. If you do not have such a GPU, you can play on your CPU. However, each turn can take a couple of minutes or more for the game to compose its response.

Playing

This README only covers installation. To learn how to play visit our Wiki.

Installation

Installing Storybro requires the following software:

  • Python 3.4 - 3.7

Grab the source code with Git and clone it to your machine:

git clone https://github.com/storybro/storybro/

Windows Installation

Storybro comes with a few Windows Batch scripts to facilitate installation. If you'd like to install manually, see Manual Installation.

Storybro uses Chocolatey, a package manager, to install dependencies:

1: Using an ADMINISTRATOR terminal from the root of this repo:

./bin/windows/install/install-choco.bat

2: Close your terminal and re-open it.

3: Now that Chocolatey is installed, install our dependencies and Storybro:

./bin/windows/install/install-storybro.bat

4: Once installation is done you should be in a shell. You can now use the storybro command:

Linux Installation

Storybro comes with a few shell scripts to faciliate installation. If you'd like to install manually, see Manual Installation.

To install Storybro simply run the install script:

./bin/linux/install/install

You can now use Poetry to enter a shell where you can use the storybro command:

poetry shell

Manual Installation

Storybro is a Python application and uses Poetry for its environment.

Install Poetry with Pip:

$ pip install poetry

Use Poetry to install Storybro:

$ poetry install

Use Poetry to run Storybro:

$ poetry run storybro

Storybro uses Aria2 to download models. Make sure that it is installed and on your $PATH if you intend to use Storybro to download models.

Community

Storybro is an open source project. Questions, discussion, and contributions are welcome. Contributions can be anything from new packages to bugfixes, documentation, or even new core features.

Resources:

Contributing

Contributing to Storybro is easy! Just send us a pull request from your fork. Make sure develop is the destination branch.

Storybro uses a rough approximation of the Git Flow branching model. The develop branch contains the latest contributions, and master is always tagged and points to the latest stable release.

If you're a contributor, make sure you're testing and playing on develop. That's where all the magic is happening (and where we hope bugs stop).

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

storybro-0.1.0.tar.gz (42.5 kB view details)

Uploaded Source

Built Distribution

storybro-0.1.0-py3-none-any.whl (48.6 kB view details)

Uploaded Python 3

File details

Details for the file storybro-0.1.0.tar.gz.

File metadata

  • Download URL: storybro-0.1.0.tar.gz
  • Upload date:
  • Size: 42.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/1.0.0 CPython/3.6.8 Linux/4.19.68

File hashes

Hashes for storybro-0.1.0.tar.gz
Algorithm Hash digest
SHA256 e684cc2e506813c69d7eae8e394641f016a11c03cd6b3889256012e3c701b09c
MD5 ffc0a03292f8d42c390a56279288705f
BLAKE2b-256 5d8c0bec65b6739388b99279325ece871fd5b5faed89c00bec8074dd036164ca

See more details on using hashes here.

File details

Details for the file storybro-0.1.0-py3-none-any.whl.

File metadata

  • Download URL: storybro-0.1.0-py3-none-any.whl
  • Upload date:
  • Size: 48.6 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/1.0.0 CPython/3.6.8 Linux/4.19.68

File hashes

Hashes for storybro-0.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 f215bac7ae33ae6af51c6d0ebc42c1bf8e0816f1bea04e8c1592e9ecc6150c56
MD5 02bc2d205cad437a3a8fe8b5045ea205
BLAKE2b-256 a05b076ed13e7a9c6cad8959a26dc19eb225b488c0260f9ef19817d8babdd99c

See more details on using hashes here.

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