From 0633a6bebdedeb840c8d1ff8e4496ecb6d653283 Mon Sep 17 00:00:00 2001 From: Muyang Li Date: Wed, 24 Apr 2024 15:17:54 -0400 Subject: [PATCH] update README --- README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/README.md b/README.md index d12f989..2d9174d 100644 --- a/README.md +++ b/README.md @@ -104,6 +104,8 @@ torchrun --nproc_per_node=$N_GPUS scripts/sdxl_example.py where `$N_GPUS` is the number GPUs you want to use. +We also provide a minimal script for running SD1.4/2 with DistriFusion in [`scripts/sd_example.py`](https://github.com/mit-han-lab/distrifuser/blob/main/scripts/sd_example.py). The usage is the same. + ### Benchmark Our benchmark results are using [PyTorch](https://pytorch.org) 2.2 and [diffusers](https://github.com/huggingface/diffusers) 0.24.0. First, you may need to install some additional dependencies: