Skip to content

Latest commit

 

History

History
26 lines (20 loc) · 449 Bytes

File metadata and controls

26 lines (20 loc) · 449 Bytes

DeepSpeed Stable Diffusion Example

Setup

Python dependencies:

pip install -r requirements.txt

Usage

Examples can be run as follows:

deepspeed --num_gpus [number of GPUs] test-[model].py

NOTE: Local CUDA graphs for replaced SD modules will only be enabled when mp_size==1.

Example Output

Command:

deepspeed --num_gpus 1 test-stable-diffusion.py

Output:

./baseline.png
./deepspeed.png