diff --git a/python/llm/example/CPU/StableDiffusion/README.md b/python/llm/example/CPU/StableDiffusion/README.md index 78598982c62..d1dcbb75a86 100644 --- a/python/llm/example/CPU/StableDiffusion/README.md +++ b/python/llm/example/CPU/StableDiffusion/README.md @@ -24,10 +24,13 @@ python ./sdxl.py Arguments info: - `--repo-id-or-model-path REPO_ID_OR_MODEL_PATH`: argument defining the huggingface repo id for the stable diffusion xl model (e.g. `stabilityai/stable-diffusion-xl-base-1.0`) to be downloaded, or the path to the huggingface checkpoint folder. It is default to be `'stabilityai/stable-diffusion-xl-base-1.0'`. -- `--prompt PROMPT`: argument defining the prompt to be infered. It is default to be `'A lovely dog on the table, detailed, 8k'`. +- `--prompt PROMPT`: argument defining the prompt to be infered. It is default to be `'An astronaut in the forest, detailed, 8k'`. - `--save-path`: argument defining the path to save the generated figure. It is default to be `sdxl-cpu.png`. - `--num-steps`: argument defining the number of inference steps. It is default to be `20`. +The sample output image looks like below. +![image](https://llm-assets.readthedocs.io/en/latest/_images/sdxl-cpu.png) + #### 4.2 LCM-LoRA Example The example shows how to performing inference with LCM-LoRA on Intel CPU. ```bash diff --git a/python/llm/example/CPU/StableDiffusion/sdxl.py b/python/llm/example/CPU/StableDiffusion/sdxl.py index ee4fc87745c..7b07731796d 100644 --- a/python/llm/example/CPU/StableDiffusion/sdxl.py +++ b/python/llm/example/CPU/StableDiffusion/sdxl.py @@ -37,7 +37,7 @@ def main(args): parser = argparse.ArgumentParser(description="Stable Diffusion") parser.add_argument('--repo-id-or-model-path', type=str, default="stabilityai/stable-diffusion-xl-base-1.0", help='The huggingface repo id for the stable diffusion model checkpoint') - parser.add_argument('--prompt', type=str, default="A lovely dog on the table, detailed, 8k", + parser.add_argument('--prompt', type=str, default="An astronaut in the forest, detailed, 8k", help='Prompt to infer') parser.add_argument('--save-path',type=str,default="sdxl-cpu.png", help="Path to save the generated figure") diff --git a/python/llm/example/GPU/StableDiffusion/README.md b/python/llm/example/GPU/StableDiffusion/README.md index 9a826329ab1..94932c3a454 100644 --- a/python/llm/example/GPU/StableDiffusion/README.md +++ b/python/llm/example/GPU/StableDiffusion/README.md @@ -97,10 +97,14 @@ python ./sdxl.py Arguments info: - `--repo-id-or-model-path REPO_ID_OR_MODEL_PATH`: argument defining the huggingface repo id for the stable diffusion xl model (e.g. `stabilityai/stable-diffusion-xl-base-1.0`) to be downloaded, or the path to the huggingface checkpoint folder. It is default to be `'stabilityai/stable-diffusion-xl-base-1.0'`. -- `--prompt PROMPT`: argument defining the prompt to be infered. It is default to be `'A lovely dog on the table, detailed, 8k'`. +- `--prompt PROMPT`: argument defining the prompt to be infered. It is default to be `'An astronaut in the forest, detailed, 8k'`. - `--save-path`: argument defining the path to save the generated figure. It is default to be `sdxl-gpu.png`. - `--num-steps`: argument defining the number of inference steps. It is default to be `20`. + +The sample output image looks like below. +![image](https://llm-assets.readthedocs.io/en/latest/_images/sdxl-gpu.png) + #### 4.2 LCM-LoRA Example The example shows how to performing inference with LCM-LoRA on Intel GPU. ```bash diff --git a/python/llm/example/GPU/StableDiffusion/sdxl.py b/python/llm/example/GPU/StableDiffusion/sdxl.py index 4ba564dc491..3b2c7ddbb5a 100644 --- a/python/llm/example/GPU/StableDiffusion/sdxl.py +++ b/python/llm/example/GPU/StableDiffusion/sdxl.py @@ -37,7 +37,7 @@ def main(args): parser = argparse.ArgumentParser(description="Stable Diffusion") parser.add_argument('--repo-id-or-model-path', type=str, default="stabilityai/stable-diffusion-xl-base-1.0", help='The huggingface repo id for the stable diffusion model checkpoint') - parser.add_argument('--prompt', type=str, default="A lovely dog on the table, detailed, 8k", + parser.add_argument('--prompt', type=str, default="An astronaut in the forest, detailed, 8k", help='Prompt to infer') parser.add_argument('--save-path',type=str,default="sdxl-gpu.png", help="Path to save the generated figure")