AI Art

AI Art is a plugin for DSS that allows you to generate images from text prompts.

Usage

For usage instructions, see the documentation.

Testing and linting

Format the code using Black:

make black

Lint the code using Flake8, and verify that the code was formatted using Black:

make lint

Run unit tests:

make unit-tests

Run integration tests:

make integration-tests

Run all linters and tests at once:

make tests

Building

Create a plugin archive that can be imported into DSS:

make plugin

Known limitations

1. Due to licensing restrictions, the weights must be acquired manually before you can use this plugin. See the documentation for details.

2. The weights must be stored on the local filesystem. If a remote folder (S3, etc) is used, or if the recipe uses containerized execution, the weights will be downloaded to a temporary directory every time the recipe is run. This is because the method used to load the weights (from_pretrained) requires a local filepath.