Skip to main content

Unified Stable Diffusion pipeline for diffusers

Project description

diffusion-ui-backend

Gradio backend for the diffusion-ui web frontend using an unified Stable Diffusion diffusers pipeline

The gradio interface provides an API to generate images with Stable Diffusion for:

  • text-to-image
  • image-to-image
  • inpainting

Documentation

The documentation is available here

Installation

Detailled installation instructions are available in the documentation.

First install pytorch with cuda support (if you have a NVIDIA GPU):

conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge

Then install diffusionui and its dependencies:

pip install diffusionui

First Usage

The first time, you have to download the model:

# using the low-memory model (for GPUs with low VRAM)
diffusionui --low-mem --download-model

# or using the full model
diffusionui --download-model

Usage

Once the model has been downloaded, you can start the backend by running:

# For the low-memory model
diffusionui --low-mem

# For the full model
diffusionui

It should produce an local URL for the gradio interface:

Running on local URL:  http://127.0.0.1:7860/

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

diffusionui-0.4.1.tar.gz (15.2 kB view hashes)

Uploaded Source

Built Distribution

diffusionui-0.4.1-py2.py3-none-any.whl (8.9 kB view hashes)

Uploaded Python 2 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