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 |
![examples](./examples.jpg)
|
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 |
![examples](./examples.jpg)
|
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" \
|