discoart - Create Disco Diffusion artworks in one line

Created at: 2022-07-01 00:25:31
Language: Python
License: NOASSERTION

Create compelling Disco Diffusion artworks in one line

PyPI Docker Cloud Build Status Open in Google Colab Twitter Follow

DiscoArt is an elegant way of creating compelling Disco Diffusion[*] artworks for generative artists, AI enthusiasts and hard-core developers. DiscoArt has a modern & professional API with a beautiful codebase, ensuring high usability and maintainability. It introduces handy features such as result recovery and persistence, gRPC/HTTP serving w/o TLS, post-analysis, easing the integration to larger cross-modal or multi-modal applications.

[*] Disco Diffusion is a Google Colab Notebook that leverages CLIP-Guided Diffusion to allow one to create compelling and beautiful images from text prompts.

👼 Available to all: fully optimized for Google Colab free tier! Perfect for those who don't own GPU by themselves.

🎨 Focus on create not code: one-liner create() with a Pythonic interface, autocompletion in IDE, and powerful features. Fetch real-time results anywhere anytime, no more worry on session outrage on Google Colab. Set initial state easily for more efficient parameter exploration.

🏭 Ready for integration & production: built on top of DocArray data structure, enjoy smooth integration with Jina, CLIP-as-service and other cross-/multi-modal applications.

Install

pip install discoart

If you are not using DiscoArt under Google Colab, then other dependencies might be required.

Get Started

Open in Google Colab

Create artworks

from discoart import create

da = create()

That's it! It will create with the default text prompts and parameters.

Set prompts and parameters

Suppported parameters are listed here. You can specify them in create():

from discoart import create

da = create(text_prompts='A painting of sea cliffs in a tumultuous storm, Trending on ArtStation.',
            init_image='https://d2vyhzeko0lke5.cloudfront.net/2f4f6dfa5a05e078469ebe57e77b72f0.png',
            skip_steps=100)

This docs explains those parameters in very details. The minor difference on the parameters between DiscoArt and DD5.x is explained here.

Visualize results

create() returns da, a DocumentArray-type object. It contains the following information:

  • All arguments passed to create() function, including seed, text prompts and model parameters.
  • The generated image and its intermediate images.

This allows you to further post-process, analyze, export the results with powerful DocArray API.

For example, you can display all final images in a grid:

da.plot_image_sprites(skip_empty=True, show_index=True, keep_aspect_ratio=True)

Note that all images perspective are preserved. You can display them one by one:

for d in da:
    d.display()

The length of da is determined by the n_batches parameter.

You can take one particular run:

da[0].display()

Images are stored as Data URI in .uri, to save it as a local file:

da[0].save_uri_to_file('discoart-result.png')

You can also zoom into a run and check out intermediate steps:

da[0].chunks.plot_image_sprites(skip_empty=True, show_index=True, keep_aspect_ratio=True)

You can .display() the chunks one by one, or save one via .save_uri_to_file().

Finally, you can review its parameters via:

da[0].tags

Pull results anywhere anytime

If you are a free-tier Google Colab user, one annoy thing is the lost of sessions from time to time. Or sometimes you just early stop the run as the first image is not good enough, and a keyboard interrupt will prevent .create() to return any result. Either case, you can easily recover the results by pulling the last session ID.

  1. Find the session ID. It appears on top of the image.

  2. Pull the result via that ID on any machine at any time, not necessarily on Google Colab:

    from docarray import DocumentArray
    
    da = DocumentArray.pull('discoart-3205998582')

Reuse a Document as the initial state

One can use a Document as the initial state for the next run. Its .tags will be used as the initial parameters; .uri if presented will be used as the initial image.

from discoart import create
from docarray import DocumentArray

da = DocumentArray.pull('discoart-3205998582')

create(init_document=da[0],
       cut_ic_pow=0.5,
       tv_scale=600, 
       cut_overview='[12]*1000', 
       cut_innercut='[12]*1000', 
       use_secondary_model=False)

Verbose logs

You can also get verbose logs by setting the following lines before import discoart:

import os

os.environ['DISCOART_LOG_LEVEL'] = 'DEBUG'

Run in Docker

Docker Image Size (tag)

We provide a prebuilt Docker image for running DiscoArt in the Jupyter Notebook.

# docker build . -t jinaai/discoart  # if you want to build yourself
docker run -p 51000:8888 -v $(pwd):/home/jovyan/ --gpus all jinaai/discoart

What's next?

Next is create.

😎 If you are already a DD user: you are ready to go! There is no extra learning, DiscoArt respects the same parameter semantics as DD5.2. So just unleash your creativity!

There are some minor differences between DiscoArt and DD5.x:

  • DiscoArt does not support video generation and image_prompt (which was marked as ineffective in DD 5.2).
  • Due to no video support, text_prompts in DiscoArt accepts a string or a list of strings, not a dictionary; i.e. no frame index 0: or 100:.
  • clip_models accepts a list of values chosen from ViT-B/32, ViT-B/16, ViT-L/14, RN101, RN50, RN50x4, RN50x16, RN50x64. Slightly different in names vs. DD5.2.

👶 If you are a DALL·E Flow user: you may want to take step by step, as Disco Diffusion works in a very different way than DALL·E. It is much more advanced and powerful: e.g. Disco Diffusion can take weighted & structured text prompts; it can initialize from a image with controlled noise; and there are way more parameters one can tweak. Impatient prompt like "armchair avocado" will give you nothing but confusion and frustration. I highly recommend you to check out the following resources before trying your own prompt:

Support

Join Us

DiscoArt is backed by Jina AI and licensed under MIT License. We are actively hiring AI engineers, solution engineers to build the next neural search ecosystem in open-source.