diff --git a/examples/05_stable_diffusion/README.md b/examples/05_stable_diffusion/README.md index aa8e3f573..8f589f70e 100644 --- a/examples/05_stable_diffusion/README.md +++ b/examples/05_stable_diffusion/README.md @@ -58,7 +58,7 @@ By default, `compile_alt.py` does not include model weights (constants) with the #### Alternative pipeline -The original pipeline requires a diffusers model local dir, and relies directly on `StableDiffusionPipeline`. This pipeline builds similar functionality without directly using `StableDiffusionPipeline`, and is capable of loading model weights from either diffusers or compvis models to compiled aitemplate modules. +The original pipeline requires a diffusers model local dir, and relies directly on `StableDiffusionPipeline`. This pipeline builds similar functionality without directly using `StableDiffusionPipeline`, and is capable of loading model weights from either diffusers or [CompVis](https://huggingface.co/CompVis) models to compiled aitemplate modules. * AITemplate modules are created * Model weights are loaded, converted/mapped, then applied to AITemplate module @@ -67,8 +67,13 @@ The original pipeline requires a diffusers model local dir, and relies directly * Loading CLIPTextModel from `ckpt` requires the appropriate `hf-hub-or-path` to be specified i.e. `runwayml/stable-diffusion-v1-5` for SD1.x checkpoints, `stabilityai/stable-diffusion-2-1` for SD2.x checkpoints. ``` -python3 scripts/demo.py --hf-hub-or-path runwayml/stable-diffusion-v1-5 --ckpt v1-5-pruned-emaonly.ckpt -python3 scripts/demo.py --hf-hub-or-path stabilityai/stable-diffusion-2-1 --ckpt v2-1_768-ema-pruned.ckpt +python3 scripts/demo_alt.py \ +--hf-hub-or-path runwayml/stable-diffusion-v1-5 \ +--ckpt v1-5-pruned-emaonly.ckpt + +python3 scripts/demo_alt.py \ +--hf-hub-or-path stabilityai/stable-diffusion-2-1 \ +--ckpt v2-1_768-ema-pruned.ckpt ``` `--ckpt` takes preference over `--hf-hub-or-path` if both are specified