sd-progress-demo icon indicating copy to clipboard operation
sd-progress-demo copied to clipboard

cheap views of intermediate Stable Diffusion results


title: Diffusers Preview Demo emoji: 📉 colorFrom: blue colorTo: blue sdk: gradio sdk_version: 3.3.1 app_file: app.py pinned: false

Here we demonstrate how to do preview images of a Stable Diffusion's intermediate stages using a fast approximation to visualize the low-resolution (64px) latent state.

  • app.py is a Gradio application that yields preview images from a generator function while the pipeline is in progress. The UI is directly derived from Stability AI's Stable Diffusion Demo.
  • progress_ipywidgets_demo.ipynb demonstrates using the same pipeline to update Jupyter widgets in a notebook.
  • preview_decoder.py has the fast latent-to-RGB decoder function.
  • generator_pipeline.py provides a DiffusionPipeline with a generate() method to yield the latent data at each step. It is nearly a strict refactoring of the StableDiffusionPipeline in 🧨diffusers 0.3.