diff options
Diffstat (limited to 'candle-examples/examples/stable-diffusion/README.md')
| -rw-r--r-- | candle-examples/examples/stable-diffusion/README.md | 22 |
1 files changed, 16 insertions, 6 deletions
diff --git a/candle-examples/examples/stable-diffusion/README.md b/candle-examples/examples/stable-diffusion/README.md index b8736a2a..feb7ca56 100644 --- a/candle-examples/examples/stable-diffusion/README.md +++ b/candle-examples/examples/stable-diffusion/README.md @@ -8,7 +8,7 @@ XL using Rust and [candle](https://github.com/huggingface/candle). The `stable-diffusion` example is a conversion of [diffusers-rs](https://github.com/LaurentMazare/diffusers-rs) using candle rather than libtorch. This implementation supports Stable Diffusion v1.5, v2.1, -as well as Stable Diffusion XL 1.0. +as well as Stable Diffusion XL 1.0, and Turbo. ## Getting the weights @@ -23,16 +23,26 @@ cargo run --example stable-diffusion --release --features=cuda,cudnn \ -- --prompt "a cosmonaut on a horse (hd, realistic, high-def)" ``` -The final image is named `sd_final.png` by default. -The default scheduler is the Denoising Diffusion Implicit Model scheduler (DDIM). The -original paper and some code can be found in the [associated repo](https://github.com/ermongroup/ddim). +The final image is named `sd_final.png` by default. The Turbo version is much +faster than previous versions, to give it a try add a `--sd-version turbo` flag, +e.g.: + +```bash +cargo run --example stable-diffusion --release --features=cuda,cudnn \ + -- --prompt "a cosmonaut on a horse (hd, realistic, high-def) --sd-version turbo" +``` + +The default scheduler for the v1.5, v2.1 and XL 1.0 version is the Denoising +Diffusion Implicit Model scheduler (DDIM). The original paper and some code can +be found in the [associated repo](https://github.com/ermongroup/ddim). +The default scheduler for the XL Turbo version is the Euler Ancestral scheduler. ### Command-line flags - `--prompt`: the prompt to be used to generate the image. - `--uncond-prompt`: the optional unconditional prompt. -- `--sd-version`: the Stable Diffusion version to use, can be `v1-5`, `v2-1`, or - `xl`. +- `--sd-version`: the Stable Diffusion version to use, can be `v1-5`, `v2-1`, + `xl`, or `turbo`. - `--cpu`: use the cpu rather than the gpu (much slower). - `--height`, `--width`: set the height and width for the generated image. - `--n-steps`: the number of steps to be used in the diffusion process. |