sd-progress-demo
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 agenerate()
method to yield the latent data at each step. It is nearly a strict refactoring of the StableDiffusionPipeline in 🧨diffusers 0.3.