Update README
Browse files
README.md
CHANGED
|
@@ -6,6 +6,7 @@ language:
|
|
| 6 |
tags:
|
| 7 |
- stable-diffusion
|
| 8 |
- stable-diffusion-xl
|
|
|
|
| 9 |
- tensorrt
|
| 10 |
- text-to-image
|
| 11 |
---
|
|
@@ -16,7 +17,10 @@ tags:
|
|
| 16 |
|
| 17 |
This repository hosts the TensorRT versions of **Stable Diffusion XL 1.0** created in collaboration with [NVIDIA](https://huggingface.co/nvidia). The optimized versions give substantial improvements in speed and efficiency.
|
| 18 |
|
| 19 |
-
See the [usage instructions](#usage-example) for how to run the SDXL pipeline with the ONNX files hosted in this repository.
|
|
|
|
|
|
|
|
|
|
| 20 |
|
| 21 |

|
| 22 |
|
|
@@ -76,6 +80,8 @@ pip3 install -r requirements.txt
|
|
| 76 |
```
|
| 77 |
|
| 78 |
4. Perform TensorRT optimized inference
|
|
|
|
|
|
|
| 79 |
```
|
| 80 |
python3 demo_txt2img_xl.py \
|
| 81 |
"Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" \
|
|
|
|
| 6 |
tags:
|
| 7 |
- stable-diffusion
|
| 8 |
- stable-diffusion-xl
|
| 9 |
+
- stable-diffusion-xl-lcm
|
| 10 |
- tensorrt
|
| 11 |
- text-to-image
|
| 12 |
---
|
|
|
|
| 17 |
|
| 18 |
This repository hosts the TensorRT versions of **Stable Diffusion XL 1.0** created in collaboration with [NVIDIA](https://huggingface.co/nvidia). The optimized versions give substantial improvements in speed and efficiency.
|
| 19 |
|
| 20 |
+
See the [usage instructions](#usage-example) for how to run the SDXL pipeline with the ONNX files hosted in this repository.
|
| 21 |
+
|
| 22 |
+
Usage instructions for the LCM version of sdxl [here](lcm/README.md)
|
| 23 |
+
|
| 24 |
|
| 25 |

|
| 26 |
|
|
|
|
| 80 |
```
|
| 81 |
|
| 82 |
4. Perform TensorRT optimized inference
|
| 83 |
+
* The first invocation produces plan files in `engine_xl_base` and `engine_xl_refiner` specific to the accelerator being run on and are reused for later invocations.
|
| 84 |
+
|
| 85 |
```
|
| 86 |
python3 demo_txt2img_xl.py \
|
| 87 |
"Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" \
|